Update build files and examples

Author: gpahal

README.md

@@ -259,7 +259,7 @@ const dgraphClient = new dgraph.DgraphClient(stub1, stub2);
 // Use dgraphClient
 // ...
 
-// Cleanup resources.
+// Cleanup resources by closing all client stubs.
 stub1.close();
 stub2.close();
 ```

examples/simple/index-pre-v7.6.js

@@ -1,8 +1,13 @@
 const dgraph = require("dgraph-js");
 const grpc = require("grpc");
 
-function newClient() {
-    const clientStub = new dgraph.DgraphClientStub("localhost:9080", grpc.credentials.createInsecure());
+// Create a client stub.
+function newClientStub() {
+    return new dgraph.DgraphClientStub("localhost:9080", grpc.credentials.createInsecure());
+}
+
+// Create a client.
+function newClient(clientStub) {
     return new dgraph.DgraphClient(clientStub);
 }
 
@@ -125,13 +130,17 @@ function queryData(dgraphClient) {
 }
 
 function main() {
-    const dgraphClient = newClient();
+    const dgraphClientStub = newClientStub();
+    const dgraphClient = newClient(dgraphClientStub);
     return dropAll(dgraphClient).then(() => {
         return setSchema(dgraphClient);
     }).then(() => {
         return createData(dgraphClient);
     }).then(() => {
         return queryData(dgraphClient);
+    }).then(() => {
+        // Close the client stub.
+        dgraphClientStub.close();
     });
 }
 

examples/simple/index.js

@@ -1,9 +1,13 @@
 const dgraph = require("dgraph-js");
 const grpc = require("grpc");
 
+// Create a client stub.
+function newClientStub() {
+    return new dgraph.DgraphClientStub("localhost:9080", grpc.credentials.createInsecure());
+}
+
 // Create a client.
-function newClient() {
-    const clientStub = new dgraph.DgraphClientStub("localhost:9080", grpc.credentials.createInsecure());
+function newClient(clientStub) {
     return new dgraph.DgraphClient(clientStub);
 }
 
@@ -114,11 +118,15 @@ async function queryData(dgraphClient) {
 }
 
 async function main() {
-    const dgraphClient = newClient();
+    const dgraphClientStub = newClientStub();
+    const dgraphClient = newClient(dgraphClientStub);
     await dropAll(dgraphClient);
     await setSchema(dgraphClient);
     await createData(dgraphClient);
     await queryData(dgraphClient);
+
+    // Close the client stub.
+    dgraphClientStub.close();
 }
 
 main().then(() => {

lib/clientStub.d.ts

@@ -9,4 +9,7 @@ export declare class DgraphClientStub {
     mutate(mu: messages.Mutation): Promise<messages.Assigned>;
     commitOrAbort(ctx: messages.TxnContext): Promise<messages.TxnContext>;
     checkVersion(check: messages.Check): Promise<messages.Version>;
+    waitForReady(deadline: grpc.Deadline): Promise<void>;
+    close(): void;
+    grpcClient(): grpc.Client;
 }

lib/clientStub.js

@@ -18,6 +18,7 @@ var DgraphClientStub = (function () {
             mutate: util_1.promisify(this.stub.mutate, this.stub),
             commitOrAbort: util_1.promisify(this.stub.commitOrAbort, this.stub),
             checkVersion: util_1.promisify(this.stub.checkVersion, this.stub),
+            waitForReady: util_1.promisify(this.stub.waitForReady, this.stub),
         };
     }
     DgraphClientStub.prototype.alter = function (op) {
@@ -35,6 +36,15 @@ var DgraphClientStub = (function () {
     DgraphClientStub.prototype.checkVersion = function (check) {
         return this.promisified.checkVersion(check);
     };
+    DgraphClientStub.prototype.waitForReady = function (deadline) {
+        return this.promisified.waitForReady(deadline);
+    };
+    DgraphClientStub.prototype.close = function () {
+        return this.stub.close();
+    };
+    DgraphClientStub.prototype.grpcClient = function () {
+        return this.stub;
+    };
     return DgraphClientStub;
 }());
 exports.DgraphClientStub = DgraphClientStub;