@hscloud/sequelize-dmdb/lib/dialects/dmdb/data-types.js

Sequelize is a promise-based Node.js ORM tool for Postgres, MySQL, MariaDB, SQLite, Microsoft SQL Server, Dameng, Amazon Redshift and Snowflake’s Data Cloud. It features solid transaction support, relations, eager and lazy loading, read replication and mo

"use strict";
const wkx = require("wkx");
const _ = require("lodash");
const momentTz = require("moment-timezone");
const moment = require("moment");
module.exports = (BaseTypes) => {
  BaseTypes.ABSTRACT.prototype.dialectTypes = "https://eco.dameng.com/docs/zh-cn/sql-dev/dmpl-sql-datatype.html";
  BaseTypes.DATE.types.dmdb = ["TIMESTAMP"];
  BaseTypes.STRING.types.dmdb = ["VAR_STRING"];
  BaseTypes.CHAR.types.dmdb = ["STRING"];
  BaseTypes.TEXT.types.dmdb = ["BLOB"];
  BaseTypes.TINYINT.types.dmdb = ["TINY"];
  BaseTypes.SMALLINT.types.dmdb = ["SHORT"];
  BaseTypes.MEDIUMINT.types.dmdb = ["INT24"];
  BaseTypes.INTEGER.types.dmdb = ["LONG"];
  BaseTypes.BIGINT.types.dmdb = ["LONGLONG"];
  BaseTypes.FLOAT.types.dmdb = ["FLOAT"];
  BaseTypes.TIME.types.dmdb = ["TIME"];
  BaseTypes.DATEONLY.types.dmdb = ["DATE"];
  BaseTypes.BOOLEAN.types.dmdb = ["TINY"];
  BaseTypes.BLOB.types.dmdb = ["TINYBLOB", "BLOB", "LONGBLOB"];
  BaseTypes.DECIMAL.types.dmdb = ["NEWDECIMAL"];
  BaseTypes.UUID.types.dmdb = false;
  BaseTypes.ENUM.types.dmdb = false;
  BaseTypes.REAL.types.dmdb = ["DOUBLE"];
  BaseTypes.DOUBLE.types.dmdb = ["DOUBLE"];
  BaseTypes.GEOMETRY.types.dmdb = ["GEOMETRY"];
  BaseTypes.JSON.types.dmdb = ["JSON"];
  class DECIMAL extends BaseTypes.DECIMAL {
    toSql() {
      let definition = super.toSql();
      if (this._unsigned) {
        definition += " UNSIGNED";
      }
      if (this._zerofill) {
        definition += " ZEROFILL";
      }
      return definition;
    }
  }
  class DATE extends BaseTypes.DATE {
    toSql() {
      return this._length ? `TIMESTAMP(${this._length})` : "TIMESTAMP";
    }
    _stringify(date, options) {
      if (!moment.isMoment(date)) {
        date = this._applyTimezone(date, options);
      }
      if (this._length) {
        return date.format(`YYYY-MM-DD HH:mm:ss.${new Array(this._length).fill("S").join("")}`);
      }
      return date.format("YYYY-MM-DD HH:mm:ss");
    }
    static parse(value, options) {
      value = value.string();
      if (value === null) {
        return value;
      }
      if (momentTz.tz.zone(options.timezone)) {
        value = momentTz.tz(value, options.timezone).toDate();
      } else {
        value = new Date(`${value} ${options.timezone}`);
      }
      return value;
    }
  }
  class DATEONLY extends BaseTypes.DATEONLY {
    static parse(value) {
      return value.string();
    }
  }
  class TIME extends BaseTypes.TIME {
    toSql() {
      return `TIME${this._length ? `(${this._length})` : ""}`;
    }
    _stringify(date, options) {
      if (!moment.isMoment(date)) {
        date = this._applyTimezone(date, options);
      }
      if (this._length) {
        return date.format(`HH:mm:ss.${new Array(this._length).fill("S").join("")}`);
      }
      return date.format("HH:mm:ss");
    }
  }
  class UUID extends BaseTypes.UUID {
    toSql() {
      return "VARCHAR2(36)";
    }
  }
  const SUPPORTED_GEOMETRY_TYPES = ["POINT", "LINESTRING", "POLYGON"];
  class GEOMETRY extends BaseTypes.GEOMETRY {
    constructor(type, srid) {
      super(type, srid);
      if (_.isEmpty(this.type)) {
        this.sqlType = this.key;
        return;
      }
      if (SUPPORTED_GEOMETRY_TYPES.includes(this.type)) {
        this.sqlType = this.type;
        return;
      }
      throw new Error(`Supported geometry types are: ${SUPPORTED_GEOMETRY_TYPES.join(", ")}`);
    }
    static parse(value) {
      value = value.buffer();
      if (!value || value.length === 0) {
        return null;
      }
      value = value.slice(4);
      return wkx.Geometry.parse(value).toGeoJSON({ shortCrs: true });
    }
    toSql() {
      return this.sqlType;
    }
  }
  class ENUM extends BaseTypes.ENUM {
    toSql(options) {
      return `ENUM(${this.values.map((value) => options.escape(value)).join(", ")})`;
    }
  }
  class JSONTYPE extends BaseTypes.JSON {
    _stringify(value, options) {
      return options.operation === "where" && typeof value === "string" ? value : JSON.stringify(value);
    }
    toSql() {
      return "TEXT";
    }
    _sanitize(value) {
      try {
        return JSON.parse(value);
      } catch (err) {
        return value;
      }
    }
  }
  JSONTYPE.parse = JSONTYPE.prototype._sanitize;
  class BOOLEAN extends BaseTypes.BOOLEAN {
    toSql() {
      return "BIT";
    }
  }
  class STRING extends BaseTypes.STRING {
    toSql() {
      if (this._binary) {
        return `VARBINARY(${this._length || 255})`;
      }
      return `VARCHAR(${this._length || 255})`;
    }
    _sanitize(value) {
      if (value instanceof Buffer) {
        return value.toString();
      }
      return value;
    }
    _stringify(value) {
      if (this._binary) {
        return Buffer.from(value).toString("hex");
      }
      return value;
    }
    _bindParam(value, options) {
      return options.bindParam(this._binary ? Buffer.from(value).toString("hex") : value);
    }
  }
  return {
    ENUM,
    DATE,
    DATEONLY,
    TIME,
    UUID,
    GEOMETRY,
    DECIMAL,
    JSON: JSONTYPE,
    BOOLEAN,
    STRING
  };
};
//# sourceMappingURL=data-types.js.map