neopg 2.2.7 → 2.2.8

package/lib/SchemaSync.js

@@ -102,9 +102,9 @@ class SchemaSync {
     for (const c of cols) inf[c.column_name] = c;
 
     await this.syncColumn(sql, def, inf, curTableName, debug, force, dropNotExistCol);
+    await this.autoRemoveIndex(sql, def, schema, tableName, debug);
     await this.syncIndex(sql, def, schema, curTableName, debug);
     await this.syncUnique(sql, def, schema, curTableName, debug);
-    await this.autoRemoveIndex(sql, def, schema, tableName, debug);
     await this.syncReferences(sql, def, schema, curTableName, schemaOid, debug, ctx, options);
 
     if (debug) console.log(` - 表结构同步完成 (${tableName}) - `);
@@ -290,6 +290,33 @@ class SchemaSync {
     const indices = def.index || [];
     if (!Array.isArray(indices)) return;
 
+    let gen_real_index = indname => {
+      let real_indexname
+      if (Array.isArray(indname)) {
+        real_indexname = indname.sort((a,b) => a.localeCompare(b)).join('_');
+      } else if (typeof indname === 'string') {
+        if (indname.indexOf(',') >= 0) {
+          real_indexname = indname.split(',').map(s=>s.trim()).filter(s=>s)
+                                .sort((a,b)=>a.localeCompare(b))
+                                .join('_');
+        } else {
+          real_indexname = indname.trim();
+        }
+      }
+
+      return real_indexname;
+    }
+
+    let unique_indexs = (def.unique || []).map(a => gen_real_index(a));
+
+    let check_in_unique = (indname) => {
+      for (let a of unique_indexs) {
+        if (a === gen_real_index(indname)) return true;
+      }
+
+      return false;
+    }
+
     for (const indname of indices) {
         const removeIndex = def.removeIndex || [];
         if (removeIndex.includes(indname)) continue;
@@ -299,7 +326,21 @@ class SchemaSync {
             continue;
         }
 
-        let idxCols = Array.isArray(indname) ? indname : indname.split(',').map(s=>s.trim()).filter(s=>s);
+        if (!Array.isArray(indname) && typeof indname !== 'string') {
+          debug && console.error(`Index ${indname} 不是字符串或数组，跳过。`);
+          continue;
+        }
+
+        if (check_in_unique(indname)) {
+          debug && console.error(`Index ${indname} 是唯一索引，不会创建index索引，跳过。`);
+          continue;
+        }
+
+        let idxCols = Array.isArray(indname)
+                     ? indname.sort((a,b) => a.localeCompare(b))
+                      : indname.split(',').map(s=>s.trim()).filter(s=>s)
+                              .sort((a,b) => a.localeCompare(b));
+
         const idxNamePart = idxCols.join('_');
         const targetIdxName = `${def.tableName}_${idxNamePart}_idx`;
 
@@ -339,7 +380,14 @@ class SchemaSync {
       for (const indname of uniques) {
           if (!this._checkColumnsExist(indname, def)) continue;
 
-          let idxCols = Array.isArray(indname) ? indname : indname.split(',').map(s=>s.trim()).filter(s=>s);
+          if (!Array.isArray(indname) && typeof indname !== 'string') {
+            debug && console.error(`Index ${indname} 不是字符串或数组，跳过。`);
+            continue;
+          }
+
+          let idxCols = Array.isArray(indname)
+                          ? indname.sort((a,b)=>a.localeCompare(b))
+                          : indname.split(',').map(s=>s.trim()).filter(s=>s).sort((a,b)=>a.localeCompare(b));
 
           // 监测是否等于主键
           if (pkSet.size > 0 && idxCols.length === pkSet.size) {
@@ -368,7 +416,7 @@ class SchemaSync {
 
   static async autoRemoveIndex(sql, def, schema, tableName, debug) {
       const allIdx = await sql`
-        SELECT indexname FROM pg_indexes 
+        SELECT * FROM pg_indexes 
         WHERE tablename = ${tableName} 
         AND schemaname = ${schema} 
         AND indexname != ${tableName + '_pkey'}
@@ -377,22 +425,51 @@ class SchemaSync {
       if (allIdx.length === 0) return;
 
       const currentIdxNames = allIdx.map(i => i.indexname);
+      let indexTable = Object.create(null);
+      for (let a of allIdx) {
+        indexTable[a.indexname] = a;
+      }
       
       const indices = def.index || [];
       const uniques = def.unique || [];
-      
-      const keepSet = new Set();
-      const makeName = (n) => `${tableName}_${Array.isArray(n) ? n.map(x=>x.trim()).join('_') : n.split(',').map(x=>x.trim()).filter(x=>x).join('_')}_idx`;
 
-      indices.forEach(n => keepSet.add(makeName(n)));
-      uniques.forEach(n => keepSet.add(makeName(n)));
+      const makeName = (n) => `${tableName}_${Array.isArray(n) 
+                        ? n.map(x=>x.trim()).sort((a,b) => a.localeCompare(b)).join('_')
+                        : n.split(',').map(x=>x.trim()).filter(x=>x).sort((a,b) => a.localeCompare(b)).join('_')}_idx`;
+
+      let allIndex = Object.create(null);
+      for (let k of indices) {
+        allIndex[makeName(k)] = k;
+      }
+
+      let allUnique = Object.create(null);
+      for (let k of uniques) {
+        allUnique[makeName(k)] = k;
+      }
 
       for (const idxName of currentIdxNames) {
-          if (!keepSet.has(idxName)) {
+          if (!allIndex[idxName] && !allUnique[idxName]) {
               if (debug) console.log('Auto removing index:', idxName);
               await sql.unsafe(`DROP INDEX ${schema}.${idxName}`);
           }
       }
+
+      for (let k in allIndex) {
+        if (!indexTable[k]) continue;
+
+        if (indexTable[k].indexdef.toLowerCase().indexOf('create index') !== 0) {
+          debug && console.log('Auto remove index with inconsistent type:', k);
+          await sql.unsafe(`DROP INDEX ${schema}.${k};`);
+        }
+      }
+
+      for (let k in allUnique) {
+        if (!indexTable[k]) continue;
+        if (indexTable[k].indexdef.toLowerCase().indexOf('create unique') !== 0) {
+          debug && console.log('Auto remove index with inconsistent type:', k);
+          await sql.unsafe(`DROP INDEX ${schema}.${k};`);
+        }
+      }
   }
 
   // --- 外键同步 ---

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "neopg",
-  "version": "2.2.7",
+  "version": "2.2.8",
   "description": "orm for postgres",
   "keywords": [
     "neopg",