This commit is contained in:
Louis Lam 2024-10-28 00:42:25 +08:00
parent 0c820a9d5b
commit 3a20e4726d
3 changed files with 60 additions and 2 deletions

View File

@ -190,6 +190,14 @@ class Database {
fs.writeFileSync(path.join(Database.dataDir, "db-config.json"), JSON.stringify(dbConfig, null, 4));
}
/**
* Get the MySQL2 Knex client
* @returns {KumaMySQL2} MySQL2 Knex client
*/
static getMySQL2KnexClient() {
return require("./utils/knex/lib/dialects/mysql2/index");
}
/**
* Connect to the database
* @param {boolean} testMode Should the connection be started in test mode?
@ -261,7 +269,7 @@ class Database {
connection.end();
config = {
client: "mysql2",
client: Database.getMySQL2KnexClient(),
connection: {
host: dbConfig.hostname,
port: dbConfig.port,
@ -284,7 +292,7 @@ class Database {
await embeddedMariaDB.start();
log.info("mariadb", "Embedded MariaDB started");
config = {
client: "mysql2",
client: Database.getMySQL2KnexClient(),
connection: {
socketPath: embeddedMariaDB.socketPath,
user: "node",

View File

@ -0,0 +1,20 @@
const ClientMySQL2 = require("knex/lib/dialects/mysql2/index");
const KumaColumnCompiler = require("./schema/mysql2-columncompiler");
/**
* Fixed MySQL2 client class.
* - Fix: Default value for TEXT fields is not supported.
*/
class KumaMySQL2 extends ClientMySQL2 {
driverName = "mysql2";
/**
*
*/
columnCompiler() {
return new KumaColumnCompiler(this, ...arguments);
}
}
module.exports = KumaMySQL2;

View File

@ -0,0 +1,30 @@
const ColumnCompilerMySQL = require("knex/lib/dialects/mysql/schema/mysql-columncompiler");
const { isObject } = require("knex/lib/util/is");
class KumaColumnCompiler extends ColumnCompilerMySQL {
/**
* Diff: Override defaultTo method to handle default value for TEXT fields
* @param {any} value Value
* @returns {string|void} Default value (Don't understand why it can return void or string, but it's the original code, lol)
*/
defaultTo(value) {
// MySQL defaults to null by default, but breaks down if you pass it explicitly
// Note that in MySQL versions up to 5.7, logic related to updating
// timestamps when no explicit value is passed is quite insane - https://dev.mysql.com/doc/refman/5.7/en/server-system-variables.html#sysvar_explicit_defaults_for_timestamp
if (value === null || value === undefined) {
return;
}
if ((this.type === "json" || this.type === "jsonb") && isObject(value)) {
// Default value for json will work only it is an expression
return `default ('${JSON.stringify(value)}')`;
}
const defaultVal = super.defaultTo.apply(this, arguments);
// louislam deleted: (this.type !== "blob" && this.type.indexOf("text") === -1)
// Other code is the same as the original code
// See: https://github.com/louislam/uptime-kuma/pull/5048#discussion_r1818076626
return defaultVal;
}
}
module.exports = KumaColumnCompiler;