mas-server 2.0.0 → 2.0.1

package/src/masmysql/masSql.ts

@@ -1,736 +0,0 @@
-import { sqlQuery } from "./esaysql";
-import state from "./state";
-import { getType } from "./utils";
-import c from "ansi-colors";
-export default class massql {
-  /** 调试模式，0为关闭，1的时候打印sql语句，2的时候打印结果 */
-  public debugMode = 0;
-  public mysql = null;
-  // ================
-  private form = null;
-  private formArr = {};
-  private formArrObj = {};
-  private lock = 0;
-  private methods = null;
-  // 查询变量
-  private selectOne = 0;
-  private page = {
-    isPaging: false,
-    pageNum: 1,
-    pageSize: 20,
-    total: 0,
-  };
-  private selectWhere: any = {};
-  private whereStr = "";
-  private keyStr = "*";
-  private jdGetTime = 0;
-  private isSort = 0;
-  private sortObj = null;
-  private sortArr = null;
-  private isDistinct = 0;
-  private distinctKey = "";
-  private countName = null;
-  private isGetdelete = 0;
-  private joinSqlStr = "";
-  // 自定义语句
-  private defineSql = null;
-  // 更新
-  private updateOBj = null;
-  private batchUpdateObj = null;
-  // 插入
-  private insertObject = null;
-  private insertArray = null;
-  constructor() {
-    if (!state.sqlForm) {
-      throw Error("请先初始化数据库!");
-    }
-    this.formArr = state.sqlForm;
-    const formArr = JSON.parse(JSON.stringify(this.formArr));
-    for (const k in formArr) {
-      const obj = {};
-      for (const item of formArr[k]) obj[item] = item;
-      formArr[k] = obj;
-    }
-    this.formArrObj = formArr;
-    this.mysql = state.connection;
-  }
-  // 条件转义
-  escape(str: string | number): string | number {
-    return state.connection.escape(str);
-  }
-  // 条件转义
-  static escape(str: string | number): string | number {
-    return state.connection.escape(str);
-  }
-  //表名转义
-  escapeId(str: string): string {
-    return state.connection.escapeId(str);
-  }
-  //表名转义
-  static escapeId(str: string): string {
-    return state.connection.escapeId(str);
-  }
-  /** 版本查看 */
-  version() {
-    sqlQuery("SELECT VERSION() AS SQL版本").then((sqlres) => {
-      console.info(c.green(sqlres[0]));
-    });
-    return this;
-  }
-  /** 字段 */
-  key(key: (string | string[])[]) {
-    this.keyStr = key
-      .map((item: any) => {
-        const t = getType(item);
-        if (t === "array") {
-          return `${this.escapeId(item[0])} AS ${this.escapeId(item[1])}`;
-        } else if (t === "string") {
-          return `${this.escapeId(item)}`;
-        } else {
-          console.info(new Error("非法的key"));
-          this.lock = 1;
-        }
-      })
-      .join(",");
-    return this;
-  }
-  /** 使用表 */
-  use(form: string | string[]) {
-    if (getType(form) == "array") {
-      form = form[form.length - 1].replace("@!", "");
-    }
-    if (!form || !this.formArr[form as any]) {
-      throw Error("表名不正确！");
-    }
-    this.form = form;
-    return this;
-  }
-  /** where条件 */
-  where(key: {} | string, value?: string | number) {
-    if (getType(key) == "string" && !value) {
-      console.info(Error("请传入value!"));
-      this.lock = 1;
-      return this;
-    }
-    if (getType(key) == "string") {
-      this.whereStr += ` AND \`${key}\` = ${this.escape(value)}`;
-    } else {
-      for (const k in key as any) {
-        this.whereStr += ` AND (\`${k}\` = ${this.escape(key[k])})`;
-      }
-    }
-    return this;
-  }
-  /** 大于等于条件 */
-  gte(key: {} | string, value?: string | number) {
-    if (getType(key) == "string" && !value) {
-      console.info(Error("请传入value!"));
-      this.lock = 1;
-      return this;
-    }
-    if (getType(key) == "string") {
-      this.whereStr += ` AND \`${key}\` >= ${this.escape(value)}`;
-    } else {
-      for (const k in key as any) {
-        this.whereStr += ` AND (\`${k}\` >= ${this.escape(key[k])})`;
-      }
-    }
-    return this;
-  }
-  /** 大于条件 */
-  gt(key: {} | string, value?: string | number) {
-    if (getType(key) == "string" && !value) {
-      console.info(Error("请传入value!"));
-      this.lock = 1;
-      return this;
-    }
-    if (getType(key) == "string") {
-      this.whereStr += ` AND \`${key}\` > ${this.escape(value)}`;
-    } else {
-      for (const k in key as any) {
-        this.whereStr += ` AND (\`${k}\` > ${this.escape(key[k])})`;
-      }
-    }
-    return this;
-  }
-  /** 小于条件 */
-  lt(key: {} | string, value?: string | number) {
-    if (getType(key) == "string" && !value) {
-      console.info(Error("请传入value!"));
-      this.lock = 1;
-      return this;
-    }
-    if (getType(key) == "string") {
-      this.whereStr += ` AND \`${key}\` < ${this.escape(value)}`;
-    } else {
-      for (const k in key as any) {
-        this.whereStr += ` AND (\`${k}\` < ${this.escape(key[k])})`;
-      }
-    }
-    return this;
-  }
-  /** 小于等于条件 */
-  lte(key: {} | string, value?: string | number) {
-    if (getType(key) == "string" && !value) {
-      console.info(Error("请传入value!"));
-      this.lock = 1;
-      return this;
-    }
-    if (getType(key) == "string") {
-      this.whereStr += ` AND \`${key}\` <= ${this.escape(value)}`;
-    } else {
-      for (const k in key as any) {
-        this.whereStr += ` AND (\`${k}\` <= ${this.escape(key[k])})`;
-      }
-    }
-    return this;
-  }
-  /** where条件，用or连接 */
-  whereOR(key: {} | string, value?: string | number) {
-    if (getType(key) == "string" && !value) {
-      console.info(Error("请传入value!"));
-      this.lock = 1;
-      return this;
-    }
-    if (getType(key) == "string") {
-      this.whereStr += `  OR \`${key}\`= ${this.escape(value)}`;
-    } else {
-      for (const k in key as any) {
-        this.whereStr += `  OR (\`${k}\` = ${this.escape(key[k])})`;
-      }
-    }
-    return this;
-  }
-  /** like条件 */
-  like(key: {} | string, value?: string | number) {
-    if (getType(key) == "string" && !value) {
-      console.info(Error("请传入value!"));
-      this.lock = 1;
-      return this;
-    }
-    if (getType(key) == "string") {
-      this.whereStr += ` AND \`${key}\` LIKE ${this.escape("%" + value + "%")}`;
-    } else {
-      for (const k in key as any) {
-        this.whereStr += ` AND (\`${k}\` LIKE ${this.escape(
-          "%" + key[k] + "%"
-        )})`;
-      }
-    }
-    return this;
-  }
-  /** like条件 */
-  likeOR(key: {} | string, value?: string | number) {
-    if (getType(key) == "string" && !value) {
-      console.info(Error("请传入value!"));
-      this.lock = 1;
-      return this;
-    }
-    if (getType(key) == "string") {
-      this.whereStr += `  OR \`${key}\` LIKE ${this.escape("%" + value + "%")}`;
-    } else {
-      for (const k in key as any) {
-        this.whereStr += `  OR (\`${k}\` LIKE ${this.escape(
-          "%" + key[k] + "%"
-        )})`;
-      }
-    }
-    return this;
-  }
-  /** in条件 */
-  in(key: string, value: string[] | number[]) {
-    if (getType(value) != "array") {
-      console.info(Error("请传入value或value格式错误! " + value));
-      this.lock = 1;
-      return this;
-    }
-    this.whereStr += ` AND (\`${key}\` in (${value
-      .map((item) => this.escape(item))
-      .join(",")}))`;
-    return this;
-  }
-  /** in条件 */
-  inOR(key: string, value?: string[] | number[]) {
-    if (getType(value) != "array") {
-      console.info(Error("请传入value或value格式错误! " + value));
-      this.lock = 1;
-      return this;
-    }
-    this.whereStr += `  OR (\`${key}\` in (${value
-      .map((item) => this.escape(item))
-      .join(",")}))`;
-    return this;
-  }
-  /** 只查一个 */
-  one() {
-    this.selectOne = 1;
-    return this;
-  }
-  /** 分页,在select时传入pageNum和pageSize */
-  paging() {
-    this.page.isPaging = true;
-    return this;
-  }
-  /** 获得时间 */
-  getTime() {
-    this.jdGetTime = 1;
-    return this;
-  }
-  /** 排序方法，mode中 DESC:降序，最新(大)的放前面 | ASC:升序，最早(小)的前面  */
-  sort(
-    obj: string | { key: string; mode: "DESC" | "ASC" }[],
-    mode: "DESC" | "ASC" = "DESC"
-  ) {
-    const objT = getType(obj);
-    if (objT == "string") {
-      this.isSort = 1;
-      this.sortObj = { key: obj, mode: mode };
-    } else if (objT == "array") {
-      this.isSort = 1;
-      this.sortArr = obj;
-    } else {
-      console.info(Error("排序参数错误！"));
-      this.lock = 1;
-    }
-    return this;
-  }
-  /** 统计数量 */
-  count(name: string) {
-    this.methods = "count";
-    if (getType(name) != "string") {
-      console.info(Error("统计参数错误！"));
-      this.lock = 1;
-      return this;
-    }
-    this.countName = name;
-    return this;
-  }
-  /** 去重 */
-  distinct(key: string) {
-    this.methods = "distinct";
-    this.isDistinct = 1;
-    const keyT = getType(key);
-    if (keyT == "string") {
-      this.distinctKey = key;
-    } else {
-      console.info(Error("去重参数错误！"));
-      this.lock = 1;
-    }
-    return this;
-  }
-  /** 连表 */
-  leftJoin(joinFormName: string, joinFormKey: string, key: string) {
-    this.joinSqlStr += `LEFT JOIN ${this.escapeId(
-      joinFormName
-    )} ON ${this.escapeId(joinFormName)}.${this.escapeId(
-      joinFormKey
-    )} = ${this.escapeId(this.form)}.${this.escapeId(key)}\n`;
-    return this;
-  }
-  /** 内连表 */
-  innerJoin(joinFormName: string, joinFormKey: string, key: string) {
-    this.joinSqlStr += `INNER JOIN ${this.escapeId(
-      joinFormName
-    )} ON ${this.escapeId(joinFormName)}.${this.escapeId(
-      joinFormKey
-    )} = ${this.escapeId(this.form)}.${this.escapeId(key)}\n`;
-    return this;
-  }
-  /** 右连表 */
-  rightJoin(joinFormName: string, joinFormKey: string, key: string) {
-    this.joinSqlStr += `RIGHT JOIN ${this.escapeId(
-      joinFormName
-    )} ON ${this.escapeId(joinFormName)}.${this.escapeId(
-      joinFormKey
-    )} = ${this.escapeId(this.form)}.${this.escapeId(key)}\n`;
-    return this;
-  }
-
-  /**
-   *
-   * @param data 为string和number时，通过id查询，为数组时通过id批量查询，为json时，通过and关联查询
-   * @param likeSearch dataObj为json时是否进行模糊查询，默认开启
-   * @param filter dataObj为json时是否对其进行过滤，默认开启
-   */
-  select(
-    dataObj?: [] | {} | string | number,
-    likeSearch: 1 | 0 = 1,
-    filter: 0 | 1 = 1
-  ) {
-    this.methods = "select";
-    if (!dataObj) {
-      return this;
-    }
-    let data: any = dataObj;
-    const dataT = getType(data);
-
-    if (dataT == "array") {
-      // 为id数组情况，把id的string都转为number
-      data = data
-        .map((item) => parseInt(item))
-        .filter((item) => !Number.isNaN(item));
-      this.whereStr += ` AND (\`id\` in (${data.join(",")}))`;
-    } else if (dataT == "string" || dataT == "number") {
-      // 为id的情况，把id的string转为number
-      this.whereStr += ` AND (\`id\`= ${parseInt(data)})`;
-    } else if (dataT == "object") {
-      // 为object的情况，默认为模糊搜索，且id不能模糊搜索
-      this.selectWhere = dataObj;
-      data = JSON.parse(JSON.stringify(data));
-      let symbol = "=";
-      let pct = "";
-      likeSearch && (symbol = "LIKE");
-      likeSearch && (pct = "%");
-      //是否进行字段过滤
-      if (filter) {
-        if (!this.form) throw Error("请把use放置在select前！");
-        for (const k in data) {
-          if (!this.formArrObj[this.form][k]) continue;
-          if (k == "id") data[k] = parseInt(data[k]);
-          this.whereStr += ` AND (\`${k}\` ${
-            k == "id" ? "=" : symbol
-          } ${this.escape(pct + data[k] + pct)})`;
-        }
-      } else {
-        for (const k in data) {
-          //不进行字段过滤，分页参数也不能要
-          if (k == "pageSize" || k == "pageNum") continue;
-          if (k == "id") data[k] = parseInt(data[k]);
-          this.whereStr += ` AND (\`${k}\` ${
-            k == "id" ? "=" : symbol
-          } ${this.escape(pct + data[k] + pct)})`;
-        }
-      }
-    }
-    return this;
-  }
-  getDelete() {
-    this.isGetdelete = 1;
-    return this;
-  }
-  /** 自定义语句 */
-  define(sqlStr) {
-    this.defineSql = sqlStr;
-    return this;
-  }
-  /** 更新 */
-  update(obj: object, id: string | number | (string | number[])) {
-    if (!obj) {
-      console.info(Error(`没有传入更新数据 ${obj}`));
-      this.lock = 1;
-    } else {
-      this.updateOBj = obj;
-    }
-    this.methods = "update";
-    const idT = getType(id);
-    if (idT == "string") {
-      id = parseInt(id as any);
-      this.where("id", id);
-    } else if (idT == "array") {
-      this.in("id", id as any);
-    } else if (idT == "number") {
-      this.where("id", id as any);
-    } else {
-      this.lock = 1;
-      console.log(Error(`更新id错误: id为 ${id}`));
-    }
-    return this;
-  }
-  /** 批量更新 */
-  batchUpdate(obj: {}[]) {
-    this.methods = "batchUpdate";
-    if (getType(obj) == "array") {
-      if (obj.length == 0) {
-        console.info(Error(`更新数据不能为空 ${obj}`));
-        this.lock = 1;
-      }
-    } else {
-      console.info(Error(`更新数据格式错误 ${obj}`));
-      this.lock = 1;
-    }
-    this.batchUpdateObj = obj;
-  }
-  /** set方法，有id则更新，无id则插入 */
-  set(obj: {} | {}[]) {
-    const objT = getType(obj);
-    if (!obj) {
-      console.info(Error(`没有传入数据 ${obj}`));
-      this.lock = 1;
-    } else if (objT == "array") {
-      if (obj[0].id) {
-        // this.update()
-        this.batchUpdate(obj as any);
-      } else {
-        this.insert(obj);
-      }
-    } else if (objT == "object") {
-      const id = (obj as any).id;
-      if (id) {
-        this.update(obj, id);
-      } else {
-        this.insert(obj);
-      }
-    } else {
-      console.info(Error(`数据格式错误 ${obj}`));
-      this.lock = 1;
-    }
-    return this;
-  }
-  /** 插入 */
-  insert(obj: {} | {}[]) {
-    this.methods = "insert";
-    const objT = getType(obj);
-    if (objT == "array") {
-      this.insertArray = obj;
-    } else if (objT == "object") {
-      this.insertObject = obj;
-    } else {
-      this.lock = 1;
-      console.info(Error(`插入数据错误 ${obj}`));
-    }
-    return this;
-  }
-  /** 删除 */
-  delete(id: string | number | (string | number)[]) {
-    this.methods = "delete";
-    const idT = getType(id);
-    if (idT == "string") {
-      id = parseInt(id as any);
-      this.where("id", id);
-    } else if (idT == "array") {
-      this.in("id", id as any);
-    } else if (idT == "number") {
-      this.where("id", id as any);
-    } else {
-      this.lock = 1;
-      console.log(Error(`更新id错误: id为 ${id}`));
-    }
-    return this;
-  }
-  buildBatchUpdateSQL(tableName, updateRecords) {
-    // 获取要更新的字段和对应的值
-    const fields = Object.keys(updateRecords[0]).filter(
-      (field) => field !== "id"
-    );
-
-    // 构建 SQL 语句
-    const sql = `UPDATE ${tableName} SET ${fields
-      .map(
-        (field) =>
-          `${this.escapeId(field)} = CASE id ${updateRecords
-            .map(
-              (record) =>
-                `WHEN ${record.id ?? "-1"} THEN ${
-                  record[field]
-                    ? this.escape(record[field])
-                    : this.escapeId(field)
-                }`
-            )
-            .join(" ")} END`
-      )
-      .join(", ")} WHERE id IN (${updateRecords
-      .map((record) => parseInt(record.id ?? "-1"))
-      .join(",")})`;
-
-    return sql;
-  }
-  private _init() {
-    // ================
-    this.form = null;
-    this.lock = 0;
-    this.methods = null;
-    // 查询变量
-    this.selectOne = 0;
-    this.page = {
-      isPaging: false,
-      pageNum: 1,
-      pageSize: 20,
-      total: 0,
-    };
-    this.selectWhere = {};
-    this.whereStr = "";
-    this.keyStr = "*";
-    this.jdGetTime = 0;
-    this.isSort = 0;
-    this.sortObj = null;
-    this.sortArr = null;
-    this.isDistinct = 0;
-    this.distinctKey = "";
-    this.isGetdelete = 0;
-    this.countName = null;
-    this.joinSqlStr = "";
-    // 自定义语句
-    this.defineSql = null;
-    // 更新
-    this.updateOBj = null;
-    this.batchUpdateObj = null;
-    // 插入
-    this.insertObject = null;
-    this.insertArray = null;
-  }
-  private batchUpdateGetSql() {
-    return this.buildBatchUpdateSQL(this.form, this.batchUpdateObj);
-  }
-  private selectGetSql() {
-    let limit = "";
-    let sort = "";
-    // 是否获取时间
-    if (this.jdGetTime && this.whereStr != "*") {
-      this.keyStr += ",`update_time`,`create_time`";
-    }
-    // 是否分页
-    if (this.page.isPaging) {
-      this.selectWhere.pageNum &&
-        (this.page.pageNum = parseInt(this.selectWhere.pageNum));
-      this.selectWhere.pageSize &&
-        (this.page.pageSize = parseInt(this.selectWhere.pageSize));
-      const start = (this.page.pageNum - 1) * this.page.pageSize;
-      limit = `LIMIT ${start},${this.page.pageSize}`;
-    }
-    // 是否排序
-    if (this.isSort == 1) {
-      if (this.sortObj) {
-        sort = `ORDER BY ${this.sortObj.key} ${this.sortObj.mode}`;
-      } else if (this.sortArr) {
-        sort = `ORDER BY ${this.sortArr
-          .map((item) => `${this.escapeId(item.key)} ${item.mode}`)
-          .join(",")}`;
-      }
-    }
-    const WHERE = this.whereStr == "" ? "" : ` AND (${this.whereStr.slice(5)})`;
-    return `SELECT ${this.keyStr} from ${this.escapeId(this.form)} ${
-      this.joinSqlStr
-    } WHERE ${
-      this.isGetdelete == 0
-        ? `${this.escapeId(this.form)}.is_delete = 0`
-        : "1=1"
-    } ${WHERE} ${limit} ${sort}`;
-  }
-  private updateGetSql() {
-    const WHERE = this.whereStr == "" ? "" : ` AND (${this.whereStr.slice(5)})`;
-    let updateStr = "";
-    for (let k in this.updateOBj) {
-      if (!this.formArrObj[this.form][k] || k == "id") continue;
-      updateStr += `${this.escapeId(k)} = ${this.escape(this.updateOBj[k])},`;
-    }
-    return `UPDATE ${this.escapeId(this.form)} SET ${updateStr.slice(
-      0,
-      updateStr.length - 1
-    )} WHERE is_delete = 0 ${WHERE};`;
-  }
-  private insertGetSql() {
-    // 只插入一个，进行字段效验
-    if (this.insertObject) {
-      const KeyStr = [];
-      const ValueStr = [];
-      for (let k in this.insertObject) {
-        if (!this.formArrObj[this.form][k]) continue;
-        KeyStr.push(this.escapeId(k));
-        ValueStr.push(this.escape(this.insertObject[k]));
-      }
-      return `INSERT INTO ${this.form} (${KeyStr.join(
-        ","
-      )}) VALUES (${ValueStr.join(",")});`;
-    } else if (this.insertArray) {
-      // 批量插入，不进行字段效验
-      const KeyStr = [];
-      const valueArr = [];
-      if (this.insertArray[0]) {
-        for (let k in this.insertArray[0]) {
-          KeyStr.push(this.escapeId(k));
-        }
-      }
-      for (let item in this.insertArray) {
-        const ValueStr = [];
-        for (let k in this.insertArray[item]) {
-          ValueStr.push(this.escape(this.insertArray[item][k]));
-        }
-        valueArr.push(ValueStr);
-      }
-      return `INSERT INTO ${this.form} (${KeyStr.join(",")}) VALUES ${valueArr
-        .map((vs) => `(${vs.join(",")})`)
-        .join(",")};`;
-    }
-  }
-  private deleteGetSql() {
-    const WHERE = this.whereStr == "" ? "" : `(${this.whereStr.slice(5)})`;
-    return `UPDATE ${this.escapeId(
-      this.form
-    )} SET is_delete = 1 where  ${WHERE};`;
-  }
-  private distinctGetSql() {
-    const WHERE = this.whereStr == "" ? "" : ` AND (${this.whereStr.slice(5)})`;
-    return `SELECT DISTINCT ${this.escapeId(this.distinctKey)} FROM ${
-      this.form
-    } WHERE ${this.isGetdelete == 0 ? `is_delete = 0` : "1=1"} ${WHERE}`;
-  }
-  private countGetSql() {
-    const WHERE = this.whereStr == "" ? "" : ` AND (${this.whereStr.slice(5)})`;
-    return `SELECT COUNT(*) AS ${this.escapeId(
-      this.countName
-    )} FROM ${this.escapeId(this.form)} WHERE ${
-      this.isGetdelete == 0 ? `is_delete = 0` : "1=1"
-    } ${WHERE};`;
-  }
-  /**
-   * 执行
-   * @param mode 0执行，-2返回sql语句，-1打印sql语句
-   */
-  async go(mode: 0 | -1 | -2 = 0) {
-    if (this.lock == 1) return null;
-    let sqlStr = "";
-    if (this.defineSql) {
-      sqlStr = this.defineSql;
-    } else {
-      if (!this.methods) {
-        throw Error("请选择方法");
-      }
-      if (!this.form) {
-        throw Error("请选择表");
-      }
-      sqlStr = this[`${this.methods}GetSql`]();
-    }
-
-    if (this.debugMode) {
-      console.info(c.greenBright("SQL: " + sqlStr));
-    }
-    if (mode == -1) {
-      this._init();
-      console.info(sqlStr);
-    } else if (mode == -2) {
-      this._init();
-      return sqlStr;
-    } else {
-      return await this.runSql(sqlStr);
-    }
-  }
-  private async runSql(sqlStr) {
-    let sqlRes = (await sqlQuery(sqlStr)) ?? null;
-
-    if (this.page.isPaging) {
-      const total = await sqlQuery(
-        `SELECT COUNT(*) as \`total\` FROM ${this.form}`
-      );
-      sqlRes = {
-        list: sqlRes,
-        pageNum: this.page.pageNum,
-        pageSize: this.page.pageSize,
-        total: total[0]["total"],
-      };
-    }
-    if (this.isDistinct == 1 && sqlRes) {
-      sqlRes = {
-        [this.distinctKey]: (sqlRes as any).map(
-          (item) => item[this.distinctKey]
-        ),
-      };
-    }
-    if (this.selectOne == 1) {
-      if (this.page.isPaging) throw Error("分页和单个查询不能同时使用！");
-      sqlRes = sqlRes[0];
-    }
-    this.debugMode === 2 && console.info(sqlRes, " SQL RESULT");
-    this._init();
-    return sqlRes;
-  }
-}