feat: query to fetch unique columns

Signed-off-by: Muhammad Aaqil <[email protected]>

Author: aaqilniz

lib/datasource.js

@@ -1316,6 +1316,40 @@ DataSource.prototype.discoverPrimaryKeys = function(modelName, options, cb) {
   return cb.promise;
 };
 
+/**
+ * Discover unique keys for a given modelName.
+ * Callback function return value is an object that can have the following properties:
+ *
+ *| Key | Type | Description |
+ *|-----|------|-------------|
+ *| owner |String | Table schema or owner (may be null). Owner defaults to current user.
+ *| tableName |String| Table name
+ *| columnName |String| Column name
+ * See [ID Properties](./Model-definition-JSON-file.html#id-properties) for more
+ * information.
+ * @param {String} modelName The model name
+ * @options {Object} options The options
+ * @param {Function} [cb] The callback function
+ * @returns {Promise} A promise with an array of Primary Keys (Property[])
+ */
+DataSource.prototype.discoverUniqueKeys = function(modelName, options, cb) {
+  this.freeze();
+
+  if (cb === undefined && typeof options === 'function') {
+    cb = options;
+    options = {};
+  }
+  options = options || {};
+  cb = cb || utils.createPromiseCallback();
+
+  if (this.connector.discoverUniqueKeys) {
+    this.connector.discoverUniqueKeys(modelName, options, cb);
+  } else if (cb) {
+    process.nextTick(cb);
+  }
+  return cb.promise;
+};
+
 /*! Method will be completely removed in LoopBack.next
 */
 /**
@@ -1627,7 +1661,8 @@ DataSource.prototype.discoverSchemas = function(tableName, options, cb) {
 
   const tasks = [
     this.discoverModelProperties.bind(this, tableName, options),
-    this.discoverPrimaryKeys.bind(this, tableName, options)];
+    this.discoverPrimaryKeys.bind(this, tableName, options),
+    this.discoverUniqueKeys.bind(this, tableName, options)];
 
   const followingRelations = options.associations || options.relations;
   if (followingRelations) {
@@ -1641,6 +1676,14 @@ DataSource.prototype.discoverSchemas = function(tableName, options, cb) {
     }
 
     const columns = results[0];
+    const uniqueKeyArray = results[2] || [];
+
+    let uniqueKeys = [];
+
+    uniqueKeyArray.forEach(key => {
+      uniqueKeys.push(key['columnName']);
+    });
+
     if (!columns || columns.length === 0) {
       cb(new Error(g.f('Table \'%s\' does not exist.', tableName)));
       return cb.promise;
@@ -1657,6 +1700,7 @@ DataSource.prototype.discoverSchemas = function(tableName, options, cb) {
       debug('Primary keys: ', pks);
     }
 
+    uniqueKeys = uniqueKeys.filter(item => !pks.hasOwnProperty(item));
     const schema = {
       name: nameMapper('table', tableName),
       options: {
@@ -1686,6 +1730,9 @@ DataSource.prototype.discoverSchemas = function(tableName, options, cb) {
       if (pks[item.columnName]) {
         schema.properties[propName].id = pks[item.columnName];
       }
+      if (uniqueKeys.includes(propName)) {
+        schema.properties[propName]['index'] = {unique: true};
+      }
       const dbSpecific = schema.properties[propName][dbType] = {
         columnName: item.columnName,
         dataType: item.dataType,