fix: Check if the SQL is valid when creating a MySQL account.

apecloud · Dec 9, 2024 · 3d12856 · 3d12856
1 parent 94e8031
commit 3d12856
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pkg/lorry/engines/mysql/user.go b/pkg/lorry/engines/mysql/user.go
@@ -114,7 +114,11 @@ func (mgr *Manager) DescribeUser(ctx context.Context, userName string) (*models.
 	return user, nil
 }
 
-func (mgr *Manager) CreateUser(ctx context.Context, userName, password, _ string) error {
+func (mgr *Manager) CreateUser(ctx context.Context, userName, password, statement string) error {
+	if !strings.HasPrefix(statement, "CREATE ") {
+		return fmt.Errorf("the user %s is created with invalid sql %s, wait and try again", userName, statement)
+	}
+
 	sql := fmt.Sprintf(createUserSQL, userName, password)
 
 	_, err := mgr.Exec(ctx, sql)