neopg 2.0.7 → 2.0.9

package/README.cn.md

@@ -196,6 +196,8 @@ db.define({
   }
 })
 
+console.log(db.has('User'))
+
 ```
 
 ### 同步数据库
@@ -228,6 +230,10 @@ const db = new NeoPG(config)
 // 注意：这是一个异步方法，因为它兼容 ESM (.mjs) 的动态导入
 await db.loadModels('./models')
 
+
+// 加载文件列表
+await db.loadFiles(['./models2/WxUser.js', './models2/Role.js'])
+
 // 加载完成后即可同步或使用
 await db.sync()
 ```

package/README.md

@@ -195,6 +195,8 @@ db.define({
   }
 })
 
+console.log(db.has('User'))
+
 ```
 
 ### Syncing Database
@@ -227,6 +229,9 @@ const db = new NeoPG(config)
 // This is asynchronous because it supports .mjs dynamic imports
 await db.loadModels('./models')
 
+//load files
+await db.loadFiles(['./models2/WxUser.js', './models2/Role.js'])
+
 // Now you can sync and use them
 await db.sync()
 ```

package/lib/ModelChain.js

@@ -6,13 +6,13 @@ const serialId = makeId.serialId
 
 // 提取常量定义
 const INT_TYPES = new Set([
-    'int', 'integer', 'smallint', 'bigint', 
+    'int', 'integer', 'smallint', 'bigint',
     'serial', 'bigserial', 'smallserial',
     'int2', 'int4', 'int8'
 ])
 
 const FLOAT_TYPES = new Set([
-    'float', 'double', 'numeric', 'decimal', 'real', 
+    'float', 'double', 'numeric', 'decimal', 'real',
     'money', 'double precision', 'float4', 'float8'
 ])
 
@@ -65,13 +65,13 @@ class ModelChain {
   clone() {
     this._ensureActive()
     const copy = new ModelChain(this.ctx, this.def, this.schema)
-    
+
     // 拷贝状态
     copy._conditions = [...this._conditions]
     copy._joins = [...this._joins]
     copy._group = [...this._group]
     copy._order = [...this._order]
-    
+
     copy._limit = this._limit
     copy._offset = this._offset
     copy._lock = this._lock
@@ -180,7 +180,7 @@ class ModelChain {
         this._conditions.push(arg1)
         return this
       }
-      
+
       if (typeof arg1 === 'object' && !Array.isArray(arg1)) {
         for (const k of Object.keys(arg1)) {
           const v = arg1[k]
@@ -266,8 +266,10 @@ class ModelChain {
 
   _buildSelectQuery() {
     const t = this.sql(this.def.tableName)
-    const c = this._columns ? this.sql(this._columns) : this.sql`*`
-    
+    const c = this._columns
+              ? ((this._columns.length===1 && this._columns[0] === '*') ? this.sql`*` : this.sql(this._columns))
+              : this.sql`*`
+
     const w = this._buildWhere()
     const o = this._buildOrder()
     const j = this._buildJoins()
@@ -277,7 +279,7 @@ class ModelChain {
     const off = this._offset ? this.sql`OFFSET ${this._offset}` : this.sql``
     const lck = this._lock || this.sql``
     const ft = this.sql`${this.sql(this.schema)}.${t}`
-    
+
     return this.sql`SELECT ${c} FROM ${ft} ${j} ${w} ${g} ${o} ${l} ${off} ${lck}`
   }
 
@@ -314,7 +316,7 @@ class ModelChain {
       const j = this._buildJoins()
       const g = this._buildGroup()
       const ft = this.sql`${this.sql(this.schema)}.${t}`
-      
+
       let countPromise
 
       if (this._group.length > 0) {
@@ -342,7 +344,7 @@ class ModelChain {
       const j = this._buildJoins()
       const g = this._buildGroup()
       const ft = this.sql`${this.sql(this.schema)}.${t}`
-      
+
       let query;
       if (this._group.length > 0) {
         query = this.sql`SELECT count(*) as total FROM (SELECT 1 FROM ${ft} ${j} ${w} ${g}) as temp`
@@ -388,17 +390,17 @@ class ModelChain {
     this._ensureActive()
     try {
       if (!data || Object.keys(data).length === 0) throw new Error('[NeoPG] Update data cannot be empty')
-      
+
         if (!this._isRaw) {
         this._prepareDataForUpdate(data)
       }
 
       if (this._conditions.length === 0) throw new Error('[NeoPG] UPDATE requires a WHERE condition')
-      
+
       const fullTable = this.sql`${this.sql(this.schema)}.${this.sql(this.def.tableName)}`
       const whereFragment = this._buildWhere()
       const retFragment = this._buildReturning()
-      
+
       const result = await this.sql`UPDATE ${fullTable} SET ${this.sql(data)} ${whereFragment} ${retFragment}`
 
       if (this._returning && this._returning.length > 0) {
@@ -418,7 +420,7 @@ class ModelChain {
       const fullTable = this.sql`${this.sql(this.schema)}.${this.sql(this.def.tableName)}`
       const whereFragment = this._buildWhere()
       const retFragment = this._buildReturning()
-        
+
       return await this.sql`DELETE FROM ${fullTable} ${whereFragment} ${retFragment}`
     } finally {
       this._destroy()
@@ -450,9 +452,9 @@ class ModelChain {
         } else {
             colFragment = this.sql(field)
         }
-        
+
         const query = this.sql`
-            SELECT ${this.sql.unsafe(func)}(${colFragment}) as val 
+            SELECT ${this.sql.unsafe(func)}(${colFragment}) as val
             FROM ${ft} ${j} ${w}
         `
 
@@ -558,4 +560,4 @@ class ModelChain {
   }
 }
 
-module.exports = ModelChain
+module.exports = ModelChain

package/lib/NeoPG.js

@@ -268,6 +268,43 @@ class NeoPG {
     return this
   }
 
+  /**
+   * 
+   * @param {string} mname 
+   */
+  has(mname) {
+    return this.registry.has(mname)
+  }
+
+  /**
+   * 
+   * @param {array} mlist 
+   */
+  async loadFiles(mlist) {
+    if (typeof mlist === 'string') {
+      mlist = [mlist]
+    }
+
+    if (!Array.isArray(mlist)) {
+      throw new Error(`[NeoPG] mlist is not [string]`)
+    }
+
+    let modelExport
+
+    for (let m of mlist) {
+      if (m.endsWith('.mjs')) {
+        const imported = await import(m)
+        modelExport = imported.default 
+      } else if (m.endsWith('.js')) {
+        modelExport = require(m)
+      } else {
+        continue
+      }
+
+      this.add(modelExport)
+    }
+  }
+
 }
 
 NeoPG.dataTypes = dataTypes

package/package.json

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

package/test/test-db.js

@@ -197,6 +197,8 @@ db.add(User)
 
   db.sync({force: true, debug: true, model: 'ShopOrder'})
 
+  console.log('test has', db.has('User'))
+
   await db.model('ShopOrder').where('1=1').delete()
   await db.model('ShopOrder').insert([
     {
@@ -256,7 +258,7 @@ db.add(User)
     console.log('update', data)
 
     let result = await tx.model('User').where(tx.sql`level > 10`).returning('*').update(data)
-    console.log(result)
+    console.log('test update returning *', result)
 
     let sex = 3
     console.log(
@@ -264,6 +266,16 @@ db.add(User)
       await tx.model('User').where(tx.sql`(sex = ${sex} or level > 10)`).select(['id', 'level', 'username', 'sex']).find()
     )
 
+    console.log(
+      'test select *',
+      await tx.model('User').select().find()
+    )
+
+    console.log(
+      'test select * count',
+      await tx.model('User').select('*').findAndCount()
+    )
+
     console.log(
       'test avg',
       await tx.model('User').avg('level')
@@ -302,4 +314,4 @@ db.add(User)
   })
 
   db.close()
-})();
+})();