Add primaryKeys() and foreignKeys() functions on Connection (#282)

* Add primaryKeys() and foreignKeys() functions on Connection

Signed-off-by: Mark Irish <[email protected]>

Author: markirish

lib/Connection.js

@@ -295,6 +295,60 @@ class Connection {
     });
   }
 
+  // TODO: Documentation
+  primaryKeys(catalog, schema, table, callback = undefined) {
+    // promise...
+    if (callback === undefined) {
+      if (!this.odbcConnection)
+      {
+        throw new Error(Connection.CONNECTION_CLOSED_ERROR);
+      }
+      return new Promise((resolve, reject) => {
+        this.odbcConnection.primaryKeys(catalog, schema, table, (error, result) => {
+          if (error) {
+            reject(error);
+          } else {
+            resolve(result);
+          }
+        });
+      });
+    }
+
+    // ...or callback
+    if (!this.odbcConnection) {
+      callback(new Error(Connection.CONNECTION_CLOSED_ERROR));
+    } else {
+      this.odbcConnection.primaryKeys(catalog, schema, table, callback);
+    }
+  }
+
+  // TODO: Documentation
+  foreignKeys(catalog, schema, table, fkCatalog, fkSchema, fkTable, callback = undefined) {
+    // promise...
+    if (callback === undefined) {
+      if (!this.odbcConnection)
+      {
+        throw new Error(Connection.CONNECTION_CLOSED_ERROR);
+      }
+      return new Promise((resolve, reject) => {
+        this.odbcConnection.foreignKeys(catalog, schema, table, fkCatalog, fkSchema, fkTable, (error, result) => {
+          if (error) {
+            reject(error);
+          } else {
+            resolve(result);
+          }
+        });
+      });
+    }
+
+    // ...or callback
+    if (!this.odbcConnection) {
+      callback(new Error(Connection.CONNECTION_CLOSED_ERROR));
+    } else {
+      this.odbcConnection.foreignKeys(catalog, schema, table, fkCatalog, fkSchema, fkTable, callback);
+    }
+  }
+
   // TODO: Documentation
   columns(catalog, schema, table, type, callback = undefined) {
     // promise...

lib/odbc.d.ts

@@ -94,7 +94,11 @@ declare namespace odbc {
 
     createStatement(callback: (error: NodeOdbcError, statement: Statement) => undefined): undefined;
 
-    tables(catalog: string, schema: string, table: string, callback: (error: NodeOdbcError, result: Result<unknown>) => undefined): undefined;
+    primaryKeys(catalog: string, schema: string, table: string, callback: (error: NodeOdbcError, result: Result<unknown>) => undefined): undefined;
+
+    foreignKeys(pkCatalog: string, pkSchema: string, pkTable: string, fkCatalog: string, fkSchema: string, fkTable: string, callback: (error: NodeOdbcError, result: Result<unknown>) => undefined): undefined;
+
+    tables(catalog: string, schema: string, table: string, type: string, callback: (error: NodeOdbcError, result: Result<unknown>) => undefined): undefined;
 
     columns(catalog: string, schema: string, table: string, callback: (error: NodeOdbcError, result: Result<unknown>) => undefined): undefined;
 
@@ -120,6 +124,10 @@ declare namespace odbc {
 
     createStatement(): Promise<Statement>;
 
+    primaryKeys(catalog: string, schema: string, table: string, callback: (error: NodeOdbcError, result: Result<unknown>) => undefined):  Promise<Result<unknown>>;
+
+    foreignKeys(pkCatalog: string, pkSchema: string, pkTable: string, fkCatalog: string, fkSchema: string, fkTable: string, callback: (error: NodeOdbcError, result: Result<unknown>) => undefined):  Promise<Result<unknown>>;
+
     tables(catalog: string, schema: string, table: string, type: string): Promise<Result<unknown>>;
 
     columns(catalog: string, schema: string, table: string, column: string): Promise<Result<unknown>>;

src/odbc.h

@@ -26,6 +26,7 @@
 #include <uv.h>
 #include <napi.h>
 #include <wchar.h>
+#include <new>
 
 #include <algorithm>
 
@@ -212,6 +213,9 @@ typedef struct StatementData {
   SQLTCHAR *catalog   = NULL;
   SQLTCHAR *schema    = NULL;
   SQLTCHAR *table     = NULL;
+  SQLTCHAR *fkCatalog = NULL;
+  SQLTCHAR *fkSchema  = NULL;
+  SQLTCHAR *fkTable   = NULL;
   SQLTCHAR *type      = NULL;
   SQLTCHAR *column    = NULL;
   SQLTCHAR *procedure = NULL;
@@ -249,6 +253,9 @@ typedef struct StatementData {
     delete[] this->catalog; this->catalog = NULL;
     delete[] this->schema; this->schema = NULL;
     delete[] this->table; this->table = NULL;
+    delete[] this->fkCatalog; this->fkCatalog = NULL;
+    delete[] this->fkSchema; this->fkSchema = NULL;
+    delete[] this->fkTable; this->fkTable = NULL;
     delete[] this->type; this->type = NULL;
     delete[] this->column; this->column = NULL;
     delete[] this->procedure; this->procedure = NULL;