Possibility to use function in multi-column index (#140)

salsita · Nov 14, 2017 · 9111821 · 9111821
1 parent cf783f1
commit 9111821
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/lib/operations/indexes.js b/lib/operations/indexes.js
@@ -10,14 +10,16 @@ function generateIndexName(table, columns, options) {
     : template`${table_name}_${_.isArray(columns) ? columns.join('_') : columns}${options.unique ? '_unique' : ''}_index`;
 }
 
+function generateColumnString(column) {
+  return /.+\(.*\)/.test(column)
+    ? column // expression
+    : template`"${column}"`; // single column
+}
+
 function generateColumnsString(columns) {
-  if (_.isArray(columns)) {
-    return columns.map(name => template`"${name}"`).join(', ');
-  }
-  if (/.+\(.*\)/.test(columns)) { // expression
-    return columns;
-  }
-  return template`"${columns}"`; // single column
+  return _.isArray(columns)
+    ? columns.map(generateColumnString).join(', ')
+    : generateColumnString(columns);
 }
 
 export const create = (table_name, columns, options = {}) => {