neopg 2.0.8 → 2.0.9

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/package.json

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

package/test/test-db.js

@@ -258,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(
@@ -266,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')
@@ -304,4 +314,4 @@ db.add(User)
   })
 
   db.close()
-})();
+})();