neopg 2.0.2 → 2.0.4

package/lib/ModelChain.js

@@ -16,12 +16,10 @@ const FLOAT_TYPES = new Set([
 ])
 
 class ModelChain {
-  constructor(ctx, def, schema = 'public') {
+  constructor(ctx, def, schema='public') {
     this.ctx = ctx
     this.def = def
-    this.sql = ctx.sql
-    
-    this.tableName = def.tableName
+    this.sql = ctx ? ctx.sql : null
     this.schema = schema
 
     // --- 查询状态 ---
@@ -35,7 +33,7 @@ class ModelChain {
     this._group = []
     this._lock = null
 
-    this._isRaw = !!def.isRaw
+    this._isRaw = false
     this._executed = false
   }
 
@@ -50,7 +48,7 @@ class ModelChain {
   _ensureActive() {
     if (this._executed) {
       throw new Error(
-        `[NeoPG] ModelChain for '${this.tableName}' has already been executed. ` +
+        `[NeoPG] ModelChain for '${this.def.tableName}' has already been executed. ` +
         `Do NOT reuse the chain variable. Use .clone() if you need to fork queries.`
       )
     }
@@ -266,7 +264,7 @@ class ModelChain {
   }
 
   _buildSelectQuery() {
-    const t = this.sql(this.tableName)
+    const t = this.sql(this.def.tableName)
     const c = this._columns ? this.sql(this._columns) : this.sql`*`
     
     const w = this._buildWhere()
@@ -310,7 +308,7 @@ class ModelChain {
       const dataQuery = this._buildSelectQuery()
 
       // 2. 总数查询
-      const t = this.sql(this.tableName)
+      const t = this.sql(this.def.tableName)
       const w = this._buildWhere()
       const j = this._buildJoins()
       const g = this._buildGroup()
@@ -338,7 +336,7 @@ class ModelChain {
   async count() {
     this._ensureActive()
     try {
-      const t = this.sql(this.tableName)
+      const t = this.sql(this.def.tableName)
       const w = this._buildWhere()
       const j = this._buildJoins()
       const g = this._buildGroup()
@@ -366,11 +364,11 @@ class ModelChain {
       const inputs = isArray ? data : [data]
       if (inputs.length === 0) throw new Error('[NeoPG] Insert data cannot be empty')
 
-      if (this.def) {
+      if (!this._isRaw) {
         this._prepareDataForInsert(inputs)
       }
 
-      const fullTable = this.sql`${this.sql(this.schema)}.${this.sql(this.tableName)}`
+      const fullTable = this.sql`${this.sql(this.schema)}.${this.sql(this.def.tableName)}`
       const retFragment = this._buildReturning()
 
       const result = await this.sql`INSERT INTO ${fullTable} ${this.sql(inputs)} ${retFragment}`
@@ -389,10 +387,14 @@ class ModelChain {
     this._ensureActive()
     try {
       if (!data || Object.keys(data).length === 0) throw new Error('[NeoPG] Update data cannot be empty')
-      if (this.def) { this._prepareDataForUpdate(data) }
+      
+        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.tableName)}`
+      const fullTable = this.sql`${this.sql(this.schema)}.${this.sql(this.def.tableName)}`
       const whereFragment = this._buildWhere()
       const retFragment = this._buildReturning()
       
@@ -412,7 +414,7 @@ class ModelChain {
     this._ensureActive()
     try {
       if (this._conditions.length === 0) throw new Error('[NeoPG] DELETE requires a WHERE condition')
-      const fullTable = this.sql`${this.sql(this.schema)}.${this.sql(this.tableName)}`
+      const fullTable = this.sql`${this.sql(this.schema)}.${this.sql(this.def.tableName)}`
       const whereFragment = this._buildWhere()
       const retFragment = this._buildReturning()
         
@@ -432,7 +434,7 @@ class ModelChain {
     try {
         if (!field) throw new Error(`[NeoPG] ${func} requires a field name.`)
 
-        const t = this.sql(this.tableName)
+        const t = this.sql(this.def.tableName)
         const w = this._buildWhere()
         const j = this._buildJoins()
         const ft = this.sql`${this.sql(this.schema)}.${t}`
@@ -524,9 +526,11 @@ class ModelChain {
       if (autoId && row[pk] === undefined) {
         row[pk] = this.def.makeId(pkLen)
       }
+
       if (make_timestamp) {
         for (const t of ts.insert) makeTimestamp(row, t)
       }
+
       for (const key in row) {
         this.def.validateField(key, row[key])
       }
@@ -538,6 +542,7 @@ class ModelChain {
     if (ts.update && ts.update.length > 0) {
       for (const t of ts.update) makeTimestamp(row, t)
     }
+
     for (const key in row) {
       this.def.validateField(key, row[key])
     }

package/lib/NeoPG.js

@@ -11,6 +11,17 @@ const path = require('node:path')
 const fs = require('node:fs')
 const { pathToFileURL } = require('node:url')
 
+/**
+ * 将下划线命名转换为大驼峰命名 (e.g. shop_order -> ShopOrder)
+ */
+function toPascalCase(str) {
+  if (!str) return ''
+
+  return str.split('_')
+    .map(word => word.charAt(0).toUpperCase() + word.slice(1))
+    .join('')
+}
+
 class NeoPG {
   constructor(config) {
     this.driver = postgres(config)
@@ -24,7 +35,9 @@ class NeoPG {
 
   table(tableName, schema = null) {
     const target = schema || this.defaultSchema
-    return new this.ModelChain(this, {tableName, isRaw: true}, target)
+    let m = new this.ModelChain(this, {tableName, isRaw: true}, target)
+    m._isRaw = true
+    return m
   }
 
   model(name, schema = null) {
@@ -32,7 +45,16 @@ class NeoPG {
     if (!item) throw new Error(`[NeoPG] Model '${name}' not found.`)
     
     const target = schema || this.defaultSchema
-    return new item.Class(this, item.def, target)
+    let m = new item.Class(this, item.def, target)
+
+    if (!m.def) {
+      m.ctx = this
+      m.sql = this.sql
+      m.def = item.def
+      m.schema = target
+    }
+
+    return m
   }
 
   // --- 注册 ---
@@ -50,6 +72,41 @@ class NeoPG {
 
     if (!rawSchema) throw new Error(`[NeoPG] Missing static schema for ${ModelClass.name}`)
 
+    // 如果没有显式指定 modelName，则尝试自动推断
+    if (!rawSchema.modelName) {
+      const className = ModelClass.name
+      
+      const genericNames = ['AnonymousModel', 'ModelChain', 'Function', '']
+
+      // 1. 优先尝试使用类名 (如果类名不是通用的)
+      if (className && !genericNames.includes(className)) {
+        rawSchema.modelName = className
+      } else if (rawSchema.tableName) {
+        rawSchema.modelName = toPascalCase(rawSchema.tableName)
+        setTimeout(() => {
+          console.error(`\x1b[33;5m[NeoPG]Warning: modelName is not specified, `
+                + `use ${rawSchema.modelName} as the modelName\x1b[0m`)
+        }, 100)
+      } else {
+        // 此时说明modelName无法确定，但是tableName也没有指定
+        throw new Error(`\x1b[31m[NeoPG] Missing modelName and tableName\x1b[0m`)
+      }
+    }
+
+    //经过以上处理，modelName已经确定了，此时若没有指定tableName则把modelName转换为小写作为tableName
+    if (!rawSchema.tableName) {
+      rawSchema.tableName = rawSchema.modelName.toLowerCase()
+
+      setTimeout(() => {
+        console.error(`\x1b[33;5m[NeoPG]Warning: tableName is not specified, `
+              + `use ${rawSchema.modelName.toLowerCase()} as the tableName\x1b[0m`)
+      }, 100)
+    }
+
+    if ((/[a-z]/).test(rawSchema.modelName[0])) {
+      throw new Error(`\x1b[31;5m[NeoPG] ${rawSchema.modelName}: modelName must start with an uppercase letter.\x1b[0m`)
+    }
+
     const def = new ModelDef(rawSchema)
 
     //已经存在又不是更新，则报错
@@ -93,6 +150,16 @@ class NeoPG {
 
     if (!options.schema) options.schema = this.defaultSchema
 
+    if (options.model) {
+      let model = this.registry.get(options.model)
+
+      if (!model) {
+        throw new Error(`[NeoPG] sync: ${options.model} not found.`)
+      }
+
+      return await SchemaSync.execute(this.driver, model.def, this, options)
+    }
+
     for (const item of this.registry.values()) {
       await SchemaSync.execute(this.driver, item.def, this, options)
     }

package/package.json

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

package/test/test-db.js

@@ -138,12 +138,79 @@ const User = {
 }
 
 // 3. 注册
-db.add(User);
+db.add(User)
 
 ;(async () => {
   await db.sync({force: true, debug: true})
   // 插入
 
+  //测试modelName和tableName
+
+  class ShopOrder extends ModelChain {
+    static schema = {
+      column: {
+        id: {
+          type: dataTypes.ID
+        },
+
+        name: {
+          type: dataTypes.STRING(30)
+        },
+
+        order_no: {
+          type: dataTypes.STRING(40)
+        }
+      }
+    }
+
+    constructor() {
+      super()
+    }
+  }
+
+  db.define(ShopOrder)
+
+  //未指定modelName
+  let Cart = {
+    tableName: 'cart',
+    column: {
+
+    }
+  }
+
+  db.define(Cart)
+
+  let Category = {
+    column: {}
+  }
+
+  try {
+    db.define(Category)
+  } catch (err) {
+    setTimeout(() => {
+      console.error(err.message)
+    }, 100)
+  }
+
+  db.sync({force: true, debug: true, model: 'ShopOrder'})
+
+  await db.model('ShopOrder').where('1=1').delete()
+  await db.model('ShopOrder').insert([
+    {
+      name: 'topbit',
+      order_no: Math.random().toString(16)
+    },
+
+    {
+      name: 'neopg',
+      order_no: Math.random().toString(16)
+    }
+  ])
+
+  console.log('get shoporder...\n',
+    await db.model('ShopOrder').where('1=1').find()
+  )
+
   await db.model('User').where('1=1').delete()
 
   try {