动态修改Knex查询中的表和连接模式

const query = knex.from('myTable1 as t1')
  .select(['myCol1'])
  .join('myTable2 as t2', 't1.myCol1', 't2.myCol2')
  .join('myTable3 as t3', 't1.myCol1', 't3.myCol3')
  .where('t1.myCol4', 'myValue');

// 期望的函数，但Knex不直接支持获取并修改join
function addSchema(query, schema) {
  query.withSchema(schema); // 仅影响from子句

  // 如何获取并修改已添加的join的schema？
  // const joins = query.getJoins(); // Knex无此API
  // for (const join of joins) {
  //   join.withSchema(schema);
  // }
}

const knex = require("knex")({ client: "mysql" }); // 示例使用MySQL客户端

const readOnlyQuery = knex
  .select("*")
  .from("#.users as u") // 使用占位符 #
  .leftJoin("#.pets as p", "u.id", "p.idUser") // JOIN中也使用占位符 #
  .where("u.id", 1);

// 建议冻结此对象，防止意外修改
Object.freeze(readOnlyQuery);

/**
 * 返回一个新的Knex QueryBuilder对象，其查询包含给定的模式。
 * @param {import('knex').QueryBuilder} queryBuilder - 带有模式占位符的基础Knex查询对象。
 * @param {String} schema - 要注入的实际数据库模式名称。
 * @returns {import('knex').Raw} - 包含指定模式的新Knex原始查询对象。
 */
function buildQueryWithSchema(queryBuilder, schema) {
  // 将基础查询转换为SQL字符串
  const sqlString = queryBuilder.toString();
  // 替换占位符。注意：Knex会为模式名添加反引号，所以我们替换的是“`#`”
  // 确保替换后的模式名也用反引号包裹，以符合SQL语法
  const newSqlString = sqlString.replaceAll("`#`", "`" + schema + "`");

  // 使用knex.raw()将修改后的SQL字符串包装成一个可执行的查询
  return knex.raw(newSqlString);
}

const queryBuilderSchemaPublic = buildQueryWithSchema(readOnlyQuery, "public");
console.log("Public Schema Query:", queryBuilderSchemaPublic.toString());

const queryBuilderSchemaPrivate = buildQueryWithSchema(readOnlyQuery, "private");
console.log("Private Schema Query:", queryBuilderSchemaPrivate.toString());

Public Schema Query: select * from `public`.`users` as `u` left join `public`.`pets` as `p` on `u`.`id` = `p`.`idUser` where `u`.`id` = 1
Private Schema Query: select * from `private`.`users` as `u` left join `private`.`pets` as `p` on `u`.`id` = `p`.`idUser` where `u`.`id` = 1

const knex = require("knex")({ client: "mysql" }); // 示例使用MySQL客户端

// 1. 定义一个带有模式占位符的基础查询
const readOnlyQuery = knex
  .select("*")
  .from("#.users as u") // 使用占位符 #
  .leftJoin("#.pets as p", "u.id", "p.idUser") // JOIN中也使用占位符 #
  .where("u.id", 1);

// 建议冻结此对象，防止意外修改其内部状态
Object.freeze(readOnlyQuery);

/**
 * 返回一个新的Knex QueryBuilder对象，其查询包含给定的模式。
 * @param {import('knex').QueryBuilder} queryBuilder - 带有模式占位符的基础Knex查询对象。
 * @param {String} schema - 要注入的实际数据库模式名称。
 * @returns {import('knex').Raw} - 包含指定模式的新Knex原始查询对象。
 */
function buildQueryWithSchema(queryBuilder, schema) {
  // 将基础查询转换为SQL字符串
  const sqlString = queryBuilder.toString();

  // 替换占位符。注意：Knex会为模式名添加反引号，所以我们替换的是“`#`”
  // 确保替换后的模式名也用反引号包裹，以符合SQL语法
  const newSqlString = sqlString.replaceAll("`#`", "`" + schema + "`");

  // 使用knex.raw()将修改后的SQL字符串包装成一个可执行的查询
  return knex.raw(newSqlString);
}

// 2. 生成模式特定的查询
const queryBuilderSchemaPublic = buildQueryWithSchema(readOnlyQuery, "public");
console.log("Public Schema Query:", queryBuilderSchemaPublic.toString());

const queryBuilderSchemaPrivate = buildQueryWithSchema(readOnlyQuery, "private");
console.log("Private Schema Query:", queryBuilderSchemaPrivate.toString());

// 实际执行查询的示例（需要配置数据库连接）
// queryBuilderSchemaPublic.then(rows => {
//   console.log("Public Schema Data:", rows);
// }).catch(err => {
//   console.error("Error executing public query:", err);
// });

// 结合UNION ALL的场景
// 假设我们有多个数据库 'db1', 'db2', 'db3'
async function createUnionQueryAcrossDatabases(databases) {
  let unionQuery = knex.raw('SELECT * FROM DUAL WHERE 1=0'); // 初始空查询，或者根据实际情况构建

  for (const dbName of databases) {
    const specificDbQuery = buildQueryWithSchema(readOnlyQuery, dbName);
    // 将每个特定数据库的查询作为UNION ALL的一部分
    // 注意：knex.unionAll()方法接受QueryBuilder或knex.raw()作为参数
    unionQuery = knex.unionAll([unionQuery, specificDbQuery]);
  }
  return unionQuery;
}

// 示例：为 'schema1' 和 'schema2' 创建联合查询
// const unionResult = await createUnionQueryAcrossDatabases(['schema1', 'schema2']);
// console.log("Union Query:", unionResult.toString());