mirror of
https://gitee.com/johng/gf
synced 2026-07-04 13:02:36 +08:00
Enhance MySQL key length compatibility and add comprehensive test for issue #4382
Co-authored-by: houseme <4829346+houseme@users.noreply.github.com>
This commit is contained in:
@ -15,6 +15,10 @@ import (
|
||||
|
||||
// DoFilter handles the sql before posts it to database.
|
||||
// This method helps handle MySQL-specific issues including key length limitations.
|
||||
//
|
||||
// For MySQL tables using utf8mb4 charset, this method automatically adds ROW_FORMAT=DYNAMIC
|
||||
// to CREATE TABLE statements to prevent "Specified key was too long; max key length is 1000 bytes" errors.
|
||||
// This is particularly important for compatibility when upgrading from older GoFrame versions.
|
||||
func (d *Driver) DoFilter(
|
||||
ctx context.Context, link gdb.Link, sql string, args []any,
|
||||
) (newSql string, newArgs []any, err error) {
|
||||
@ -35,17 +39,26 @@ func (d *Driver) DoFilter(
|
||||
func (d *Driver) handleMySQLKeyLengthCompatibility(sql string) string {
|
||||
// For CREATE TABLE statements with utf8mb4 charset, ensure key length compatibility
|
||||
// This helps prevent "Specified key was too long; max key length is 1000 bytes" errors
|
||||
if strings.Contains(strings.ToUpper(sql), "CREATE TABLE") &&
|
||||
strings.Contains(strings.ToLower(sql), "utf8mb4") {
|
||||
sqlUpper := strings.ToUpper(sql)
|
||||
sqlLower := strings.ToLower(sql)
|
||||
|
||||
if strings.Contains(sqlUpper, "CREATE TABLE") &&
|
||||
(strings.Contains(sqlLower, "utf8mb4") || strings.Contains(sqlLower, "charset=utf8mb4")) {
|
||||
// Add ROW_FORMAT=DYNAMIC to enable larger key prefixes when using utf8mb4
|
||||
if !strings.Contains(strings.ToUpper(sql), "ROW_FORMAT") {
|
||||
if !strings.Contains(sqlUpper, "ROW_FORMAT") {
|
||||
// Insert ROW_FORMAT=DYNAMIC before ENGINE clause if it exists
|
||||
if strings.Contains(strings.ToUpper(sql), "ENGINE=") {
|
||||
if strings.Contains(sqlUpper, "ENGINE=") {
|
||||
sql = strings.Replace(sql, "ENGINE=", "ROW_FORMAT=DYNAMIC ENGINE=", 1)
|
||||
} else if strings.Contains(sqlUpper, "ENGINE ") {
|
||||
// Handle case where there's a space after ENGINE
|
||||
sql = strings.Replace(sql, "ENGINE ", "ROW_FORMAT=DYNAMIC ENGINE ", 1)
|
||||
} else {
|
||||
// Append ROW_FORMAT=DYNAMIC at the end of CREATE TABLE statement
|
||||
sql = strings.TrimSuffix(sql, ";")
|
||||
sql = strings.TrimSuffix(strings.TrimSpace(sql), ";")
|
||||
sql += " ROW_FORMAT=DYNAMIC"
|
||||
if !strings.HasSuffix(sql, ";") {
|
||||
sql += ";"
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@ -32,6 +32,7 @@ func Test_Issue4382_KeyLengthLimit(t *testing.T) {
|
||||
t.AssertNil(err)
|
||||
|
||||
// This should not fail with key length error in GoFrame 2.9
|
||||
// Our DoFilter enhancement should automatically add ROW_FORMAT=DYNAMIC
|
||||
_, err = db.Exec(ctx, longTableSQL)
|
||||
if err != nil {
|
||||
// If we get the specific key length error, this confirms the issue
|
||||
|
||||
Reference in New Issue
Block a user