提交供应商以及删除无关代码
This commit is contained in:
364
config/dbv2.js
364
config/dbv2.js
@@ -1,13 +1,15 @@
|
||||
// queryBuilder.js
|
||||
class QueryBuilder {
|
||||
constructor() {
|
||||
this.conditions = {};
|
||||
this.conditions = []; // { sql, params }
|
||||
this.limit = null;
|
||||
this.offset = null;
|
||||
this.groupBy = null;
|
||||
this.orderBy = null;
|
||||
}
|
||||
|
||||
where(condition, ...params) {
|
||||
this.conditions[condition] = params;
|
||||
this.conditions.push({ sql: condition, params });
|
||||
return this;
|
||||
}
|
||||
|
||||
@@ -26,88 +28,48 @@ class QueryBuilder {
|
||||
return this;
|
||||
}
|
||||
|
||||
sqdata(sql, params) {
|
||||
return new Promise((resolve, reject) => {
|
||||
global.sqlReq.query(sql, params, (err, result) => {
|
||||
if (err) {
|
||||
reject(err);
|
||||
}
|
||||
resolve(result);
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
getParams() {
|
||||
return Object.values(this.conditions).flat();
|
||||
orderByField(field, direction = 'ASC') {
|
||||
this.orderBy = { field, direction: direction.toUpperCase() };
|
||||
return this;
|
||||
}
|
||||
|
||||
buildConditions() {
|
||||
return Object.keys(this.conditions).map(condition => `${condition}`).join(' AND ');
|
||||
if (!this.conditions.length) return { sql: '', params: [] };
|
||||
const sql = this.conditions.map(c => `(${c.sql})`).join(' AND ');
|
||||
const params = this.conditions.flatMap(c => c.params);
|
||||
return { sql, params };
|
||||
}
|
||||
|
||||
buildLimitOffset() {
|
||||
let sql = '';
|
||||
if (this.limit !== null) sql += ` LIMIT ${this.limit}`;
|
||||
if (this.offset !== null) sql += ` OFFSET ${this.offset}`;
|
||||
return sql;
|
||||
}
|
||||
|
||||
async execute(db) {
|
||||
const { sql, params } = this.build();
|
||||
const [rows] = await db.query(sql, params);
|
||||
return rows;
|
||||
}
|
||||
}
|
||||
|
||||
// ------------------- SELECT -------------------
|
||||
class SelectBuilder extends QueryBuilder {
|
||||
constructor() {
|
||||
super();
|
||||
this.selectFields = [];
|
||||
this.tables = [];
|
||||
this.orderByField = '';
|
||||
this.orderByDirection = 'ASC';
|
||||
this.subQueries = []; // 用于存储子查询
|
||||
this.unions = []; // 存储UNION查询
|
||||
}
|
||||
// 添加UNION查询
|
||||
union(queryBuilder, type = 'UNION') {
|
||||
this.unions.push({queryBuilder, type});
|
||||
return this;
|
||||
}
|
||||
|
||||
// 添加UNION ALL查询
|
||||
unionAll(queryBuilder) {
|
||||
this.union(queryBuilder, 'UNION ALL');
|
||||
return this;
|
||||
}
|
||||
|
||||
// 构建主查询部分（不含ORDER BY/LIMIT/OFFSET）
|
||||
buildMainQuery() {
|
||||
const subQuerySQL = this.subQueries.map(({alias, subQuery}) => `(${subQuery}) AS ${alias}`);
|
||||
const selectClause = this.selectFields.concat(subQuerySQL).join(', ');
|
||||
|
||||
let sql = `SELECT ${selectClause}
|
||||
FROM ${this.tables.join(' ')}`;
|
||||
|
||||
const conditionClauses = this.buildConditions();
|
||||
if (conditionClauses) {
|
||||
sql += ` WHERE ${conditionClauses}`;
|
||||
}
|
||||
|
||||
if (this.groupBy) {
|
||||
sql += ` GROUP BY ${this.groupBy}`;
|
||||
}
|
||||
|
||||
const params = this.getParams();
|
||||
return {sql, params};
|
||||
}
|
||||
|
||||
// 供UNION查询调用的构建方法
|
||||
buildForUnion() {
|
||||
return this.buildMainQuery();
|
||||
this.joins = [];
|
||||
this.unions = [];
|
||||
}
|
||||
|
||||
select(fields) {
|
||||
this.selectFields = fields.split(',').map(field => field.trim());
|
||||
return this;
|
||||
}
|
||||
|
||||
// 添加子查询
|
||||
addSubQuery(alias, subQuery) {
|
||||
this.subQueries.push({alias, subQuery});
|
||||
return this;
|
||||
}
|
||||
|
||||
whereLike(fields, keyword) {
|
||||
const likeConditions = fields.map(field => `${field} LIKE ?`).join(' OR ');
|
||||
this.conditions[likeConditions] = fields.map(() => `%${keyword}%`);
|
||||
if (typeof fields === 'string') {
|
||||
this.selectFields = fields.split(',').map(f => f.trim());
|
||||
} else if (Array.isArray(fields)) {
|
||||
this.selectFields = fields;
|
||||
}
|
||||
return this;
|
||||
}
|
||||
|
||||
@@ -116,78 +78,76 @@ class SelectBuilder extends QueryBuilder {
|
||||
return this;
|
||||
}
|
||||
|
||||
leftJoin(table, condition) {
|
||||
this.tables.push(`LEFT JOIN ${table} ON ${condition}`);
|
||||
join(type, table, condition) {
|
||||
this.joins.push(`${type.toUpperCase()} JOIN ${table} ON ${condition}`);
|
||||
return this;
|
||||
}
|
||||
|
||||
orderBy(field, direction = 'ASC') {
|
||||
this.orderByField = field;
|
||||
this.orderByDirection = direction.toUpperCase();
|
||||
union(queryBuilder, type = 'UNION') {
|
||||
this.unions.push({ queryBuilder, type });
|
||||
return this;
|
||||
}
|
||||
|
||||
paginate(page, pageSize) {
|
||||
unionAll(queryBuilder) {
|
||||
return this.union(queryBuilder, 'UNION ALL');
|
||||
}
|
||||
|
||||
build() {
|
||||
const selectClause = this.selectFields.length ? this.selectFields.join(', ') : '*';
|
||||
let sql = `SELECT ${selectClause} FROM ${this.tables.join(', ')}`;
|
||||
if (this.joins.length) sql += ' ' + this.joins.join(' ');
|
||||
|
||||
const { sql: whereSql, params } = this.buildConditions();
|
||||
if (whereSql) sql += ` WHERE ${whereSql}`;
|
||||
|
||||
if (this.groupBy) sql += ` GROUP BY ${this.groupBy}`;
|
||||
|
||||
// 处理 UNION
|
||||
for (const u of this.unions) {
|
||||
const { sql: uSql, params: uParams } = u.queryBuilder.build();
|
||||
sql += ` ${u.type} ${uSql}`;
|
||||
params.push(...uParams);
|
||||
}
|
||||
|
||||
if (this.orderBy) {
|
||||
sql += ` ORDER BY ${this.orderBy.field} ${this.orderBy.direction}`;
|
||||
}
|
||||
|
||||
sql += this.buildLimitOffset();
|
||||
return { sql, params };
|
||||
}
|
||||
|
||||
async paginateWithCount(db, page = 1, pageSize = 10) {
|
||||
if (page <= 0 || pageSize <= 0) {
|
||||
throw new Error('分页参数必须大于0');
|
||||
}
|
||||
this.limit = pageSize;
|
||||
this.offset = (page - 1) * pageSize;
|
||||
return this;
|
||||
}
|
||||
|
||||
async chidBuild() {
|
||||
// count 查询
|
||||
let countSql = `SELECT COUNT(*) as total FROM ${this.tables.join(', ')}`;
|
||||
if (this.joins.length) countSql += ' ' + this.joins.join(' ');
|
||||
|
||||
let sql = `SELECT ${this.selectFields.join(', ')}
|
||||
FROM ${this.tables.join(' ')}`;
|
||||
let conditionClauses = this.buildConditions();
|
||||
if (conditionClauses) {
|
||||
sql += ` WHERE ${conditionClauses}`;
|
||||
}
|
||||
if (this.orderByField) {
|
||||
sql += ` ORDER BY ${this.orderByField} ${this.orderByDirection}`;
|
||||
}
|
||||
if (this.limit !== null) {
|
||||
sql += ` LIMIT ${this.limit}`;
|
||||
}
|
||||
if (this.offset !== null) {
|
||||
sql += ` OFFSET ${this.offset}`;
|
||||
}
|
||||
return sql;
|
||||
}
|
||||
const { sql: whereSql, params } = this.buildConditions();
|
||||
if (whereSql) countSql += ` WHERE ${whereSql}`;
|
||||
|
||||
async build() {
|
||||
const main = this.buildMainQuery();
|
||||
let fullSql = `(${main.sql})`;
|
||||
const allParams = [...main.params];
|
||||
const [countRows] = await db.query(countSql, params);
|
||||
const total = countRows[0].total;
|
||||
|
||||
// 处理UNION部分
|
||||
for (const union of this.unions) {
|
||||
const unionBuilder = union.queryBuilder;
|
||||
if (!(unionBuilder instanceof SelectBuilder)) {
|
||||
throw new Error('UNION query must be a SelectBuilder instance');
|
||||
}
|
||||
const unionResult = unionBuilder.buildForUnion();
|
||||
fullSql += ` ${union.type} (${unionResult.sql})`;
|
||||
allParams.push(...unionResult.params);
|
||||
}
|
||||
// 数据查询
|
||||
this.setLimit(pageSize).setOffset((page - 1) * pageSize);
|
||||
const { sql, params: dataParams } = this.build();
|
||||
const [rows] = await db.query(sql, dataParams);
|
||||
|
||||
// 添加ORDER BY、LIMIT、OFFSET
|
||||
if (this.orderByField) {
|
||||
fullSql += ` ORDER BY ${this.orderByField} ${this.orderByDirection}`;
|
||||
}
|
||||
if (this.limit !== null) {
|
||||
fullSql += ` LIMIT ${this.limit}`;
|
||||
}
|
||||
if (this.offset !== null) {
|
||||
fullSql += ` OFFSET ${this.offset}`;
|
||||
}
|
||||
console.log(fullSql,allParams);
|
||||
return await this.sqdata(fullSql, allParams);
|
||||
return {
|
||||
data: rows,
|
||||
total,
|
||||
page,
|
||||
pageSize,
|
||||
totalPages: Math.ceil(total / pageSize)
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
// ------------------- UPDATE -------------------
|
||||
class UpdateBuilder extends QueryBuilder {
|
||||
constructor() {
|
||||
super();
|
||||
@@ -201,157 +161,107 @@ class UpdateBuilder extends QueryBuilder {
|
||||
}
|
||||
|
||||
set(field, value) {
|
||||
if (value && value.increment && typeof value === 'object' ) {
|
||||
this.updateFields[field] = {increment: value.increment};
|
||||
if (typeof value === 'object' && value.increment !== undefined) {
|
||||
this.updateFields[field] = { increment: value.increment };
|
||||
} else {
|
||||
this.updateFields[field] = value;
|
||||
}
|
||||
return this;
|
||||
}
|
||||
|
||||
async build() {
|
||||
let sql = `UPDATE ${this.table}
|
||||
SET `;
|
||||
let updateClauses = Object.keys(this.updateFields).map(field => {
|
||||
const value = this.updateFields[field];
|
||||
if (value && value.increment && typeof value === 'object' ) {
|
||||
setFields(fieldsObj) {
|
||||
Object.entries(fieldsObj).forEach(([k, v]) => this.set(k, v));
|
||||
return this;
|
||||
}
|
||||
|
||||
build() {
|
||||
const updateClauses = Object.keys(this.updateFields).map(field => {
|
||||
const val = this.updateFields[field];
|
||||
if (typeof val === 'object' && val.increment !== undefined) {
|
||||
return `${field} = ${field} + ?`;
|
||||
}
|
||||
return `${field} = ?`;
|
||||
}).join(', ');
|
||||
|
||||
sql += updateClauses;
|
||||
let sql = `UPDATE ${this.table} SET ${updateClauses}`;
|
||||
const { sql: whereSql, params: whereParams } = this.buildConditions();
|
||||
if (whereSql) sql += ` WHERE ${whereSql}`;
|
||||
|
||||
let conditionClauses = this.buildConditions();
|
||||
if (conditionClauses) {
|
||||
sql += ` WHERE ${conditionClauses}`;
|
||||
}
|
||||
// 处理参数，确保自增字段也传入增量值
|
||||
const params = [
|
||||
...Object.values(this.updateFields).map(value =>
|
||||
(value && value.increment && typeof value === 'object' ) ? value.increment : value
|
||||
...Object.values(this.updateFields).map(v =>
|
||||
(typeof v === 'object' && v.increment !== undefined) ? v.increment : v
|
||||
),
|
||||
...this.getParams()
|
||||
...whereParams
|
||||
];
|
||||
return await this.sqdata(sql, params);
|
||||
return { sql, params };
|
||||
}
|
||||
}
|
||||
|
||||
// ------------------- INSERT -------------------
|
||||
class InsertBuilder extends QueryBuilder {
|
||||
constructor() {
|
||||
super();
|
||||
this.table = '';
|
||||
this.insertValues = [];
|
||||
this.records = [];
|
||||
this.updateValues = {};
|
||||
}
|
||||
|
||||
insertInto(table) {
|
||||
into(table) {
|
||||
this.table = table;
|
||||
return this;
|
||||
}
|
||||
|
||||
// 仍然保留单条记录的插入
|
||||
values(values) {
|
||||
if (Array.isArray(values)) {
|
||||
this.insertValues = values;
|
||||
} else {
|
||||
this.insertValues = [values]; // 将单条记录包装成数组
|
||||
}
|
||||
values(records) {
|
||||
if (!Array.isArray(records)) records = [records];
|
||||
this.records = records;
|
||||
return this;
|
||||
}
|
||||
|
||||
// 新增方法，支持一次插入多条记录
|
||||
valuesMultiple(records) {
|
||||
if (!Array.isArray(records) || records.length === 0) {
|
||||
throw new Error('Values must be a non-empty array');
|
||||
}
|
||||
|
||||
// 确保每一条记录都是对象
|
||||
records.forEach(record => {
|
||||
if (typeof record !== 'object') {
|
||||
throw new Error('Each record must be an object');
|
||||
}
|
||||
});
|
||||
|
||||
this.insertValues = records;
|
||||
upsert(records, updateFields) {
|
||||
this.values(records);
|
||||
this.updateValues = updateFields;
|
||||
return this;
|
||||
}
|
||||
|
||||
// 新增 upsert 方法，支持更新或插入
|
||||
upsert(values, updateFields) {
|
||||
// values: 要插入的记录
|
||||
// updateFields: 如果记录存在时，需要更新的字段
|
||||
if (!Array.isArray(values) || values.length === 0) {
|
||||
throw new Error('Values must be a non-empty array');
|
||||
build() {
|
||||
if (!this.records.length) throw new Error('No values to insert');
|
||||
|
||||
const columns = Object.keys(this.records[0]);
|
||||
const placeholders = `(${columns.map(() => '?').join(', ')})`;
|
||||
const valuesClause = this.records.map(() => placeholders).join(', ');
|
||||
const params = this.records.flatMap(r => columns.map(c => r[c]));
|
||||
|
||||
let sql = `INSERT INTO ${this.table} (${columns.join(', ')}) VALUES ${valuesClause}`;
|
||||
|
||||
if (Object.keys(this.updateValues).length) {
|
||||
const updates = Object.keys(this.updateValues)
|
||||
.map(f => `${f} = VALUES(${f})`) // MySQL 8.0+ 兼容写法
|
||||
.join(', ');
|
||||
sql += ` ON DUPLICATE KEY UPDATE ${updates}`;
|
||||
}
|
||||
|
||||
// 检查每条记录是否是对象
|
||||
values.forEach(record => {
|
||||
if (typeof record !== 'object') {
|
||||
throw new Error('Each record must be an object');
|
||||
}
|
||||
});
|
||||
|
||||
this.insertValues = values;
|
||||
this.updateValues = updateFields || {};
|
||||
return this;
|
||||
}
|
||||
|
||||
async build() {
|
||||
if (this.insertValues.length === 0) {
|
||||
throw new Error("No values to insert");
|
||||
}
|
||||
|
||||
// 获取表单列名，假设所有记录有相同的字段
|
||||
const columns = Object.keys(this.insertValues[0]);
|
||||
|
||||
// 构建 VALUES 子句，支持批量插入
|
||||
const valuePlaceholders = this.insertValues.map(() =>
|
||||
`(${columns.map(() => '?').join(', ')})`
|
||||
).join(', ');
|
||||
|
||||
// 展平所有的插入值
|
||||
const params = this.insertValues.flatMap(record =>
|
||||
columns.map(column => record[column])
|
||||
);
|
||||
|
||||
// 如果有 updateFields，构建 ON DUPLICATE KEY UPDATE 子句
|
||||
let updateClause = '';
|
||||
if (Object.keys(this.updateValues).length > 0) {
|
||||
updateClause = ' ON DUPLICATE KEY UPDATE ' +
|
||||
Object.keys(this.updateValues).map(field => {
|
||||
return `${field} = VALUES(${field})`;
|
||||
}).join(', ');
|
||||
}
|
||||
|
||||
// 生成 SQL 语句
|
||||
const sql = `INSERT INTO ${this.table} (${columns.join(', ')})
|
||||
VALUES ${valuePlaceholders} ${updateClause}`;
|
||||
// 执行查询
|
||||
return await this.sqdata(sql, params);
|
||||
return { sql, params };
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
// ------------------- DELETE -------------------
|
||||
class DeleteBuilder extends QueryBuilder {
|
||||
constructor() {
|
||||
super();
|
||||
this.table = '';
|
||||
}
|
||||
|
||||
deleteFrom(table) {
|
||||
from(table) {
|
||||
this.table = table;
|
||||
return this;
|
||||
}
|
||||
|
||||
async build() {
|
||||
let sql = `DELETE
|
||||
FROM ${this.table}`;
|
||||
let conditionClauses = this.buildConditions();
|
||||
if (conditionClauses) {
|
||||
sql += ` WHERE ${conditionClauses}`;
|
||||
}
|
||||
return await this.sqdata(sql, this.getParams());
|
||||
build() {
|
||||
let sql = `DELETE FROM ${this.table}`;
|
||||
const { sql: whereSql, params } = this.buildConditions();
|
||||
if (!whereSql) throw new Error('DELETE without WHERE is not allowed!');
|
||||
sql += ` WHERE ${whereSql}`;
|
||||
return { sql, params };
|
||||
}
|
||||
}
|
||||
|
||||
@@ -359,5 +269,5 @@ module.exports = {
|
||||
SelectBuilder,
|
||||
UpdateBuilder,
|
||||
InsertBuilder,
|
||||
DeleteBuilder,
|
||||
DeleteBuilder
|
||||
};
|
||||
|
||||
Reference in New Issue
Block a user