wok-server 0.5.0 → 0.6.0

package/src/mysql/manager/ops/find.ts

@@ -0,0 +1,226 @@
+import { PoolConnection, RowDataPacket } from 'mysql2'
+import { Table } from '../../table-info'
+import { MixCriteria, buildQuery } from './criteria'
+import { promiseQuery } from '../utils'
+import { MysqlConfig } from '../../config'
+
+/**
+ * 按 id 查询
+ * @param connection
+ * @param mapping
+ * @param type
+ * @param id
+ */
+export async function findById<T>(
+  config: MysqlConfig,
+  connection: PoolConnection,
+  table: Table<T>,
+  id: string | number
+): Promise<T | null> {
+  const res = await promiseQuery(config, connection, `select * from ?? where ?? = ?`, [
+    table.tableName,
+    table.id,
+    id
+  ])
+  const rows = res as RowDataPacket[]
+  if (!rows.length) {
+    return null
+  }
+  return rows[0] as T
+}
+
+/**
+ * 查询表中所有数据。需谨慎操作，全表查询有可能会产生大列表问题。
+ * @param connection
+ * @param mapping
+ * @param type
+ * @returns
+ */
+export async function findAll<T>(
+  config: MysqlConfig,
+  connection: PoolConnection,
+  table: Table<T>
+): Promise<T[]> {
+  const res = await promiseQuery(config, connection, `select * from ??`, [table.tableName])
+  return res as RowDataPacket[] as T[]
+}
+
+export interface FindOpts<T> {
+  /**
+   * 表
+   */
+  table: Table<T>
+  /**
+   * 查询条件
+   * @param criteria
+   * @returns
+   */
+  criteria?: MixCriteria<T>
+  /**
+   * 偏移，需要 limit 有值才可以
+   */
+  offset?: number
+  /**
+   * 限制返回的数量
+   */
+  limit?: number
+  /**
+   * 排序规则，按先后顺序放入，每个规则是一个元组，第一个元素是字段名称，第二个元素是顺序
+   */
+  orderBy?: Array<[keyof T, 'asc' | 'desc']>
+}
+/**
+ * 条件查询
+ * @param opts
+ * @returns
+ */
+export async function find<T>(
+  config: MysqlConfig,
+  conn: PoolConnection,
+  opts: FindOpts<T>
+): Promise<T[]> {
+  let query = opts.criteria ? buildQuery(opts.criteria) : undefined
+  const values: any[] = []
+  let sql = `select * from ?? `
+  values.push(opts.table.tableName)
+  if (query) {
+    sql += ` where ${query.sql} `
+    values.push(...query.values)
+  }
+  // 排序
+  if (opts.orderBy && opts.orderBy.length) {
+    opts.orderBy.forEach((orderBy, idx) => {
+      const [field, sort] = orderBy
+      if (idx == 0) {
+        sql += ` order by ?? ${sort} `
+      } else {
+        sql += ` , ?? ${sort} `
+      }
+      values.push(field)
+    })
+  }
+  // 数量限制
+  if (opts.limit) {
+    sql += ` limit ${opts.limit} `
+    if (opts.offset) {
+      sql += ` offset ${opts.offset}`
+    }
+  }
+  const res = await promiseQuery(config, conn, sql, values)
+  return res as RowDataPacket[] as T[]
+}
+
+/**
+ * 指定字段条件查询选项
+ */
+export interface FindSelectOpts<T, K extends keyof T> extends FindOpts<T> {
+  /**
+   * 要查询的字段
+   */
+  select: K[]
+}
+
+/**
+ * 查询指定的字段，返回值类型与选择的字段有关
+ * @param onfig
+ * @param conn
+ * @param opts
+ */
+export async function findSelect<T, K extends keyof T>(
+  config: MysqlConfig,
+  conn: PoolConnection,
+  opts: FindSelectOpts<T, K>
+): Promise<Array<Pick<T, K>>> {
+  let query = opts.criteria ? buildQuery(opts.criteria) : undefined
+  const values: any[] = []
+  let sql = `select ${opts.select.map(() => '??').join(',')} from ?? `
+  values.push(...opts.select, opts.table.tableName)
+  if (query) {
+    sql += ` where ${query.sql} `
+    values.push(...query.values)
+  }
+  // 排序
+  if (opts.orderBy && opts.orderBy.length) {
+    opts.orderBy.forEach((orderBy, idx) => {
+      const [field, sort] = orderBy
+      if (idx == 0) {
+        sql += ` order by ?? ${sort} `
+      } else {
+        sql += ` , ?? ${sort} `
+      }
+      values.push(field)
+    })
+  }
+  // 数量限制
+  if (opts.limit) {
+    sql += ` limit ${opts.limit} `
+    if (opts.offset) {
+      sql += ` offset ${opts.offset}`
+    }
+  }
+  const res = await promiseQuery(config, conn, sql, values)
+  return res as RowDataPacket[] as Pick<T, K>[]
+}
+/**
+ * 根据 id 列表查询记录
+ * @param connection
+ * @param table
+ * @param ids
+ * @returns
+ */
+export async function findByIdIn<T>(
+  config: MysqlConfig,
+  connection: PoolConnection,
+  table: Table<T>,
+  ids: Array<string | number>
+): Promise<T[]> {
+  if (!ids.length) {
+    return []
+  }
+  return find(config, connection, {
+    table,
+    criteria: c => c.in(table.id, ids)
+  })
+}
+
+/**
+ * 查找符合条件的第一条记录
+ * @param conn
+ * @param table
+ * @param criteria
+ * @returns
+ */
+export async function findFirst<T>(
+  config: MysqlConfig,
+  conn: PoolConnection,
+  table: Table<T>,
+  criteria?: MixCriteria<T>,
+  /**
+   * 排序规则，按先后顺序放入，每个规则是一个元组，第一个元素是字段名称，第二个元素是顺序
+   */
+  orderBy?: Array<[keyof T, 'asc' | 'desc']>
+): Promise<T | null> {
+  let query = criteria ? buildQuery(criteria) : undefined
+  let sql = `select * from ?? `
+  const values: any[] = [table.tableName]
+  if (query) {
+    sql += ` where ${query.sql} `
+    values.push(...query.values)
+  }
+  // 排序
+  if (orderBy && orderBy.length) {
+    orderBy.forEach((orderBy, idx) => {
+      const [field, sort] = orderBy
+      if (idx == 0) {
+        sql += ` order by ?? ${sort} `
+      } else {
+        sql += ` , ?? ${sort} `
+      }
+      values.push(field)
+    })
+  }
+  sql += ' limit 1'
+  const res = await promiseQuery(config, conn, sql, values)
+  const list = res as RowDataPacket[] as T[]
+  return list.length ? list[0] : null
+}

package/src/mysql/manager/ops/index.ts

@@ -0,0 +1,11 @@
+// 各类操作封装
+export * from './count'
+export * from './criteria'
+export * from './delete'
+export * from './exist'
+export * from './find'
+export * from './insert'
+export * from './modify'
+export * from './paginate'
+export * from './query'
+export * from './update'

package/src/mysql/manager/ops/insert.ts

@@ -0,0 +1,120 @@
+import { PoolConnection, ResultSetHeader } from 'mysql2'
+import { MysqlException } from '../../exception'
+import { Table } from '../../table-info'
+import { promiseQuery } from '../utils'
+import { MysqlConfig } from '../../config'
+import { processColumnValue } from './utils'
+
+/**
+ * 为表插入数据
+ * @param connection
+ * @param table
+ * @param data
+ * @returns
+ */
+export async function insert<T>(
+  config: MysqlConfig,
+  connection: PoolConnection,
+  table: Table<T>,
+  data: T
+): Promise<T> {
+  // 插入后的新数据
+  // 列信息，使用 set 防止 columns 中重复配置 id 和更新创建时间列
+  let columnsSet: Set<keyof T> = new Set()
+  // 判定下 id ，如果有值，才在 insert 语句中出现 id 列，否则不出现
+  if (data[table.id]) {
+    columnsSet.add(table.id)
+  }
+  table.columns.forEach(col => columnsSet.add(col))
+  if (table.createdDate) {
+    const createdData = table.createdDate.type === 'date' ? new Date() : new Date().getTime()
+    data[table.createdDate.column] = createdData as any
+    columnsSet.add(table.createdDate.column)
+  }
+  if (table.updatedDate) {
+    const updatedDate = table.updatedDate.type === 'date' ? new Date() : new Date().getTime()
+    data[table.updatedDate.column] = updatedDate as any
+    columnsSet.add(table.updatedDate.column)
+  }
+  const columns = Array.from(columnsSet)
+  // 构建 sql
+  const sql = `insert into ??(${columns.map(() => '??').join(',')}) values(${columns
+    .map(() => '?')
+    .join(',')})`
+  const values: any[] = [
+    table.tableName,
+    ...columns,
+    ...columns.map(col => processColumnValue(data[col]))
+  ]
+  const res = await promiseQuery(config, connection, sql, values)
+  const packet = res as ResultSetHeader
+  if (packet.affectedRows !== 1) {
+    throw new MysqlException(
+      `Insert failed，table：${table.tableName}，primary key: ${data[table.id]}`
+    )
+  }
+  // 自动生成的id处理
+  if (packet.insertId && (data[table.id] === undefined || data[table.id] === null)) {
+    data[table.id] = packet.insertId as any
+  }
+  return data
+}
+/**
+ * 一次插入多条记录
+ * @param connection 连接
+ * @param table 表
+ * @param list 要插入的记录列表
+ */
+export async function insertMany<T>(
+  config: MysqlConfig,
+  connection: PoolConnection,
+  table: Table<T>,
+  list: T[]
+): Promise<void> {
+  if (!list.length) {
+    return
+  }
+  // 列信息，使用 set 防止 columns 中重复配置 id 和更新创建时间列
+  let columnsSet: Set<keyof T> = new Set()
+  // 批量插入必须统一使用 id 或不使用 id，以第一条记录为准
+  if (list[0][table.id]) {
+    columnsSet.add(table.id)
+  }
+  table.columns.forEach(col => columnsSet.add(col))
+  let createdData: Date | number | undefined = undefined
+  if (table.createdDate) {
+    createdData = table.createdDate.type === 'date' ? new Date() : new Date().getTime()
+    columnsSet.add(table.createdDate.column)
+  }
+  let updatedDate: Date | number | undefined = undefined
+  if (table.updatedDate) {
+    updatedDate = table.updatedDate.type === 'date' ? new Date() : new Date().getTime()
+    columnsSet.add(table.updatedDate.column)
+  }
+  const columns = Array.from(columnsSet)
+  // 构建 sql
+  let sql = `insert into ??(${columns.map(() => '??').join(',')}) values`
+  const values: any[] = [table.tableName, ...columns]
+  list.forEach((data, idx) => {
+    if (idx > 0) {
+      sql += ','
+    }
+    sql += `(${columns.map(() => '?').join(',')})`
+    if (table.createdDate) {
+      data[table.createdDate.column] = createdData as any
+    }
+    if (table.updatedDate) {
+      data[table.updatedDate.column] = updatedDate as any
+    }
+    values.push(...columns.map(col => processColumnValue(data[col])))
+  })
+
+  const res = await promiseQuery(config, connection, sql, values)
+  const rsh = res as ResultSetHeader
+  if (rsh.affectedRows !== list.length) {
+    throw new MysqlException(
+      `Insert many for table ${table.tableName} failed,expected to insert ${list.length} records, actually ${rsh.affectedRows}.`
+    )
+  }
+  // insert 插入多条记录是不会返回生成的 id 的，所以就没有办法自动处理
+}

package/src/mysql/manager/ops/modify.ts

@@ -0,0 +1,14 @@
+import { PoolConnection, ResultSetHeader } from 'mysql2'
+import { promiseQuery } from '../utils'
+import { MysqlConfig } from '../../config'
+
+export async function modify(
+  config: MysqlConfig,
+  connection: PoolConnection,
+  sql: string,
+  values?: any[]
+): Promise<number> {
+  const res = await promiseQuery(config, connection, sql, values || [])
+  const pck = res as ResultSetHeader
+  return pck.affectedRows
+}

package/src/mysql/manager/ops/paginate.ts

@@ -0,0 +1,60 @@
+import { PoolConnection } from 'mysql2'
+import { Table } from '../../table-info'
+import { count } from './count'
+import { MixCriteria } from './criteria'
+import { find } from './find'
+import { MysqlConfig } from '../../config'
+
+export interface MysqlPaginateOpts<T> {
+  /**
+   * 表
+   */
+  table: Table<T>
+  /**
+   * 查询条件
+   * @param criteria
+   * @returns
+   */
+  criteria?: MixCriteria<T>
+  /**
+   * 而码,默认1
+   */
+  pn?: number
+  /**
+   * 每页的数据量大小,默认 20, 必须在 1-1000 之间
+   */
+  pz?: number
+  /**
+   * 排序规则，按先后顺序放入，每个规则是一个元组，第一个元素是字段名称，第二个元素是顺序
+   */
+  orderBy?: Array<[keyof T, 'asc' | 'desc']>
+}
+/**
+ * mysql 分页查询结果
+ */
+export interface MysqlPage<T> {
+  total: number
+  list: T[]
+}
+
+export async function paginate<T>(
+  config: MysqlConfig,
+  conn: PoolConnection,
+  opts: MysqlPaginateOpts<T>
+): Promise<MysqlPage<T>> {
+  const pn = opts.pn && opts.pn >= 1 ? opts.pn : 1
+  const limit = opts.pz && opts.pz >= 1 && opts.pz <= 1000 ? opts.pz : 20
+  const offset = (pn - 1) * limit
+  const list = await find(config, conn, {
+    table: opts.table,
+    criteria: opts.criteria,
+    offset,
+    limit,
+    orderBy: opts.orderBy
+  })
+  const total = await count(config, conn, opts.table, opts.criteria)
+  return {
+    total,
+    list
+  }
+}

package/src/mysql/manager/ops/query.ts

@@ -0,0 +1,13 @@
+import { PoolConnection, RowDataPacket } from 'mysql2'
+import { promiseQuery } from '../utils'
+import { MysqlConfig } from '../../config'
+
+export async function query<T>(
+  config: MysqlConfig,
+  connection: PoolConnection,
+  sql: string,
+  values?: any[]
+): Promise<T[]> {
+  const res = await promiseQuery(config, connection, sql, values || [])
+  return res as RowDataPacket[] as T[]
+}