使用Node.js操作SQLite插入数据时的主键校验问题

首先，你的校验函数存在一个核心问题：Node.js中的SQLite操作是异步的，你在DataBase.get的回调里返回false，但这个返回值并不会传递到外层的Validation函数中——外层函数会在回调执行前就已经结束，所以你根本拿不到预期的布尔结果。

接下来分两部分解决你的问题：先修复异步校验函数，再给出避免使用SELECT语句的方案。

一、修复异步校验函数（支持返回布尔值与查询数据） #

我们可以用Promise封装SQLite操作，配合async/await来让代码更易读，同时正确返回结果。这个方案既可以检查主键是否存在，也能返回对应的行数据：

// 封装成Promise的校验函数
async function checkPrimaryKeyExists(db, userEntry) {
  console.log('Validation...');
  try {
    const sql = "SELECT * FROM UsersTable WHERE My_PRIMARY_KEY = ?";
    const row = await db.get(sql, [userEntry]);
    // 如果row存在，说明主键已存在，返回true和行数据；否则返回false和null
    return { exists: !!row, data: row };
  } catch (err) {
    console.error(err.message);
    throw err; // 抛出错误让调用者处理
  }
}

// 使用示例
async function main() {
  const db = /* 你的SQLite数据库实例 */;
  const userInput = "用户输入的主键值";
  const result = await checkPrimaryKeyExists(db, userInput);
  
  if (result.exists) {
    console.log("主键已存在，对应数据：", result.data);
  } else {
    console.log("主键不存在，可以插入新数据");
    // 这里执行插入操作
  }
}

main().catch(err => console.error(err));

这个函数返回的对象包含exists（布尔值，标识主键是否存在）和data（查询到的行数据，主键存在时才会有值），完美满足你后续返回相关信息的需求。

二、避免使用SELECT语句的方案：利用SQLite主键约束 #

如果你想完全避免提前用SELECT查询主键是否存在，可以直接尝试插入数据，利用SQLite的主键唯一性约束来捕获冲突错误：

SQLite会在插入重复主键时抛出错误，错误信息包含UNIQUE constraint failed。我们可以捕获这个特定错误来判断主键是否已存在：

async function insertNewUser(db, userData) {
  try {
    const sql = "INSERT INTO UsersTable (My_PRIMARY_KEY, /* 其他字段 */) VALUES (?, ?)";
    const result = await db.run(sql, [userData.primaryKey, /* 其他字段值 */]);
    console.log("插入成功，影响行数：", result.changes);
    return true;
  } catch (err) {
    if (err.message.includes("UNIQUE constraint failed")) {
      console.log("主键已存在，无法插入");
      return false;
    } else {
      console.error("插入失败：", err.message);
      throw err;
    }
  }
}

// 使用示例
async function main() {
  const db = /* 你的SQLite数据库实例 */;
  const newUser = {
    primaryKey: "用户输入的主键",
    // 其他字段...
  };
  
  const insertSuccess = await insertNewUser(db, newUser);
  if (!insertSuccess) {
    // 如果需要，可以在这里再查询已存在的用户数据
    const existingUser = await db.get("SELECT * FROM UsersTable WHERE My_PRIMARY_KEY = ?", [newUser.primaryKey]);
    console.log("已存在的用户数据：", existingUser);
  }
}

main().catch(err => console.error(err));

另外，你也可以使用INSERT OR IGNORE语句：这条语句会在主键存在时忽略插入操作，不会抛出错误，你可以通过result.changes来判断是否成功插入（changes为1表示插入成功，0表示主键已存在）：

async function insertOrIgnoreUser(db, userData) {
  const sql = "INSERT OR IGNORE INTO UsersTable (My_PRIMARY_KEY, /* 其他字段 */) VALUES (?, ?)";
  const result = await db.run(sql, [userData.primaryKey, /* 其他字段值 */]);
  
  if (result.changes === 1) {
    console.log("插入成功");
    return { success: true, exists: false };
  } else {
    console.log("主键已存在，未插入");
    // 可选：查询已存在的数据
    const existingData = await db.get("SELECT * FROM UsersTable WHERE My_PRIMARY_KEY = ?", [userData.primaryKey]);
    return { success: false, exists: true, data: existingData };
  }
}

总结 #

• 异步操作必须用Promise/async-await处理，否则无法正确获取返回值；
• 如果要完全避免提前SELECT，直接尝试插入并捕获UNIQUE约束错误，或使用INSERT OR IGNORE；
• 若需要返回已存在的用户数据，在判断主键存在后再执行一次SELECT即可（这时候的查询是必要的，但只在冲突时触发，性能影响很小）。

内容的提问来源于stack exchange，提问作者YetiTheDev