init: add workspace files
This commit is contained in:
200
data/db.js
Normal file
200
data/db.js
Normal file
@@ -0,0 +1,200 @@
|
||||
const Database = require("better-sqlite3");
|
||||
const path = require("path");
|
||||
const fs = require("fs").promises;
|
||||
const crypto = require("crypto");
|
||||
|
||||
const stmtCaches = new Map();
|
||||
|
||||
class SqliteDatabase {
|
||||
db_path = null;
|
||||
constructor() {
|
||||
this.db = null;
|
||||
}
|
||||
|
||||
// 获取预编译语句（带缓存）
|
||||
getStmt(sql) {
|
||||
let stmt = stmtCaches.get(sql);
|
||||
if (!stmt) {
|
||||
stmt = this.db.prepare(sql);
|
||||
stmtCaches.set(sql, stmt);
|
||||
}
|
||||
return stmt;
|
||||
}
|
||||
|
||||
// 初始化数据库
|
||||
async init(db_name) {
|
||||
try {
|
||||
this.db_path = path.join(__dirname, db_name);
|
||||
const fileExists = await this.checkDbFileExists();
|
||||
|
||||
if (!fileExists) {
|
||||
console.log("数据库文件不存在，正在创建...");
|
||||
this.db = new Database(this.db_path);
|
||||
await this.executeDefaultScripts();
|
||||
console.log("数据库初始化完成");
|
||||
} else {
|
||||
console.log("数据库文件已存在，正在连接...");
|
||||
this.db = new Database(this.db_path);
|
||||
await this.executeAlterScripts();
|
||||
console.log("数据库连接成功");
|
||||
}
|
||||
} catch (error) {
|
||||
console.error("数据库初始化失败:", error);
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
|
||||
// 检查数据库文件是否存在
|
||||
async checkDbFileExists() {
|
||||
try {
|
||||
await fs.access(this.db_path);
|
||||
return true;
|
||||
} catch (error) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
// 执行 ALTER 脚本（数据库已存在时）
|
||||
async executeAlterScripts() {
|
||||
for (let sql of ALTER_SCRIPTS) {
|
||||
try {
|
||||
await this.query(sql);
|
||||
} catch (error) {
|
||||
console.error(sql, "查询失败", error);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 执行默认建表脚本（数据库新建时）
|
||||
async executeDefaultScripts() {
|
||||
for (let sql of DEFAULT_TABLE_SCRIPTS) {
|
||||
try {
|
||||
await this.query(sql);
|
||||
} catch (error) {
|
||||
console.error(sql, "查询失败", error);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 执行 INSERT/UPDATE/DELETE 等操作
|
||||
query(sql, params = []) {
|
||||
return new Promise((resolve, reject) => {
|
||||
try {
|
||||
const stmt = this.getStmt(sql);
|
||||
const info = stmt.run(...params);
|
||||
resolve({ lastID: info.lastInsertRowid, changes: info.changes });
|
||||
} catch (err) {
|
||||
console.error("SQL执行错误:", err);
|
||||
reject(err);
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
// 查询单行数据
|
||||
get(sql, params = []) {
|
||||
return new Promise((resolve, reject) => {
|
||||
try {
|
||||
const stmt = this.getStmt(sql);
|
||||
const row = stmt.get(...params);
|
||||
resolve(row);
|
||||
} catch (err) {
|
||||
console.error("SQL查询错误:", err);
|
||||
reject(err);
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
// 查询多行数据
|
||||
all(sql, params = []) {
|
||||
return new Promise((resolve, reject) => {
|
||||
try {
|
||||
const stmt = this.getStmt(sql);
|
||||
const rows = stmt.all(...params);
|
||||
resolve(rows);
|
||||
} catch (err) {
|
||||
console.error("SQL查询错误:", err);
|
||||
reject(err);
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
// 关闭数据库连接
|
||||
close() {
|
||||
return new Promise((resolve, reject) => {
|
||||
try {
|
||||
if (this.db) {
|
||||
this.db.close();
|
||||
this.db = null;
|
||||
}
|
||||
resolve();
|
||||
} catch (err) {
|
||||
console.error("关闭数据库连接失败:", err);
|
||||
reject(err);
|
||||
}
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
// 导出单例
|
||||
module.exports = new SqliteDatabase();
|
||||
|
||||
// ---------- 建表及初始脚本 ----------
|
||||
const DEFAULT_TABLE_SCRIPTS = [
|
||||
`CREATE TABLE IF NOT EXISTS users (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
name VARCHAR(60) NOT NULL,
|
||||
pwd VARCHAR(60) NOT NULL,
|
||||
phone VARCHAR(20),
|
||||
state INTEGER DEFAULT 1,
|
||||
level INTEGER DEFAULT 0,
|
||||
create_time TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||
UNIQUE (name)
|
||||
)`,
|
||||
`CREATE TABLE IF NOT EXISTS servers (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
name VARCHAR(60) NOT NULL,
|
||||
ip VARCHAR(60) NOT NULL,
|
||||
port VARCHAR(60) NOT NULL,
|
||||
state INTEGER DEFAULT 1,
|
||||
isdef INTEGER DEFAULT 0,
|
||||
istest INTEGER DEFAULT 0,
|
||||
create_time TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||
start_time TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
||||
)`,
|
||||
`CREATE TABLE IF NOT EXISTS players (
|
||||
id VARCHAR(40) NOT NULL,
|
||||
name VARCHAR(30) NOT NULL,
|
||||
userid INTEGER NOT NULL,
|
||||
sid INTEGER NOT NULL,
|
||||
level INTEGER DEFAULT 0,
|
||||
title VARCHAR(30) NOT NULL,
|
||||
data TEXT,
|
||||
create_time TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||
update_time TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
||||
)`,
|
||||
`CREATE TABLE IF NOT EXISTS players_bak (
|
||||
id VARCHAR(40) NOT NULL,
|
||||
name VARCHAR(30) NOT NULL,
|
||||
userid INTEGER NOT NULL,
|
||||
sid INTEGER NOT NULL,
|
||||
level INTEGER DEFAULT 0,
|
||||
title VARCHAR(30) NOT NULL,
|
||||
data TEXT,
|
||||
create_time TIMESTAMP,
|
||||
update_time TIMESTAMP
|
||||
)`,
|
||||
`CREATE UNIQUE INDEX IF NOT EXISTS idx_users_name ON users (name)`,
|
||||
`CREATE UNIQUE INDEX IF NOT EXISTS idx_players_id ON players (id)`,
|
||||
`CREATE INDEX IF NOT EXISTS idx_players_userid ON players (userid)`,
|
||||
`CREATE UNIQUE INDEX IF NOT EXISTS idx_players_name ON players (name)`,
|
||||
`CREATE UNIQUE INDEX IF NOT EXISTS idx_players_bak_id ON players_bak (id)`,
|
||||
`INSERT OR IGNORE INTO users(id,name,pwd,level) VALUES(1,'administrator','${MD5("123456")}',6)`,
|
||||
];
|
||||
|
||||
function MD5(str) {
|
||||
let md5 = crypto.createHash("md5");
|
||||
let result = md5.update(str + process.env.MD5_PREFIX).digest("hex");
|
||||
return result.toUpperCase();
|
||||
}
|
||||
|
||||
const ALTER_SCRIPTS = [];
|
||||
1
data/def/data.js
Normal file
1
data/def/data.js
Normal file
@@ -0,0 +1 @@
|
||||
{ }
|
||||
132
data/sql.js
Normal file
132
data/sql.js
Normal file
@@ -0,0 +1,132 @@
|
||||
const db = require('./db');
|
||||
|
||||
|
||||
module.exports = {
|
||||
|
||||
connect: async function (path) {
|
||||
await db.init(path);
|
||||
},
|
||||
close: function () {
|
||||
return db.close();
|
||||
},
|
||||
getUserBy: function (type, value) {
|
||||
return this.getUser(type + "=?", [value]);
|
||||
},
|
||||
getUserByID: function (id) {
|
||||
return this.getUser("id=?", [id]);
|
||||
},
|
||||
getUser: function (cond, paras) {
|
||||
let sql = "select id,name,pwd,phone,state,level from users";
|
||||
|
||||
if (cond) {
|
||||
sql += " where " + cond;
|
||||
}
|
||||
return db.get(sql, paras);
|
||||
},
|
||||
getUsers: function (cond, paras) {
|
||||
let sql = "select id,name,pwd,phone,state,create_time from users";
|
||||
|
||||
if (cond) {
|
||||
sql += " where " + cond;
|
||||
}
|
||||
return db.all(sql, paras);
|
||||
},
|
||||
checkUserName: function (name) {
|
||||
let sql = "select name,phone from users where name=?";// or phone=?
|
||||
return db.get(sql, [name]);
|
||||
},
|
||||
updateUser: async function (user) {
|
||||
let sql = "update users set name=?,phone=?,pwd=? where id=?";
|
||||
let result = await db.query(sql, [user.name, user.phone, user.pwd, user.id]);
|
||||
return result.changes === 1;
|
||||
},
|
||||
createUser: async function (user) {
|
||||
|
||||
let result = await db.query("insert into users(name,pwd,phone) values(?,?,?)",
|
||||
[user.name, user.pwd, user.phone]);
|
||||
if (!(result.lastID > 0)) return false;
|
||||
user.id = result.lastID;
|
||||
return true;
|
||||
},
|
||||
|
||||
|
||||
|
||||
getServers: function () {
|
||||
let sql = "select id,name, ip,port,istest,isdef,create_time,start_time from servers";
|
||||
|
||||
return db.all(sql);
|
||||
},
|
||||
getServer: function (id) {
|
||||
let sql = "select id,name, ip,port,istest,isdef from servers where id=?";
|
||||
|
||||
return db.get(sql, [id]);
|
||||
},
|
||||
saveServer: async function (data) {
|
||||
let sql = "update servers set name=?, ip=?,port=?,istest=?,isdef=? where id=?";
|
||||
|
||||
let result = await db.query(sql, [data.name, data.ip, data.port, data.istest, data.isdef, data.id]);
|
||||
|
||||
return result.changes === 1;
|
||||
},
|
||||
addServer: async function (data) {
|
||||
let sql = "insert into servers(name,ip,port,istest,isdef) values(?,?,?,?,?)";
|
||||
let result = await db.query(sql,
|
||||
[data.name, data.ip, data.port, data.istest, data.isdef]);
|
||||
|
||||
return result.lastID;
|
||||
|
||||
},
|
||||
deleteServer: function (id) {
|
||||
return db.query("delete from servers where id =?", [id]);
|
||||
},
|
||||
|
||||
getRoles: function (uid, server) {
|
||||
return db.all("select a.id,a.name,a.title,a.level from players a where a.userid=? and a.sid=?",
|
||||
[uid, server]);
|
||||
|
||||
},
|
||||
addRole: function (role) {
|
||||
|
||||
return db.query("insert into players(userid,id,name,title,level,sid,data) values(?,?,?,?,?,?,?)",
|
||||
[role.userid, role.id, role.name, role.title, role.level, role.server, role.data]);
|
||||
|
||||
},
|
||||
deleteRole: async function (userid, roleid) {
|
||||
|
||||
let sql = "insert into players_bak(id,name,userid,title,level,sid,data,create_time,update_time) select id,name,userid,title,level,sid,data,create_time,update_time from players where id=? and userid=?";
|
||||
let result = await db.query(sql, [roleid, userid]);
|
||||
if (result.changes != 1)
|
||||
return false;
|
||||
sql = "delete from players where id=? and userid=?";
|
||||
result = await db.query(sql, [roleid, userid]);
|
||||
if (result.changes != 1)
|
||||
return false;
|
||||
return true;
|
||||
|
||||
},
|
||||
saveRole: function (role) {
|
||||
return db.query("update players set name=?,title=?,level=?,data=? where userid=? and id=?",
|
||||
[role.name, role.title, role.level, role.data, role.userid, role.id]);
|
||||
},
|
||||
exitsRoleName: function (name) {
|
||||
let sql = "select name from players where name=?";// or phone=?
|
||||
return db.get(sql, [name]);
|
||||
},
|
||||
|
||||
getData: function (id) {
|
||||
return db.get("select a.id,a.name,a.title,a.level,a.data,b.pwd,b.level user_level from players a left join users b on a.userid=b.id where a.userid=?",
|
||||
[id]);
|
||||
},
|
||||
updateRoleName: function (id, name) {
|
||||
|
||||
WORLD_DATA.query_db("update players set name=? where id=? ",
|
||||
[name, id]);
|
||||
|
||||
},
|
||||
updateUserid: function (id, fromuserid, touserid) {
|
||||
WORLD_DATA.query_db("update player set userid=? where id=? and userid=?",
|
||||
[touserid, id, fromuserid]);
|
||||
}
|
||||
|
||||
};
|
||||
|
||||
Reference in New Issue
Block a user