update: 优化lang参数设置,自动转换为小写字符串

hanchuanchuan · Apr 3, 2020 · ec01295 · ec01295
1 parent d5bb6cc
commit ec01295
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/session/session_inception.go b/session/session_inception.go
@@ -307,6 +307,8 @@ func (s *session) ExecuteInc(ctx context.Context, sql string) (recordSets []sqle
 		}
 	}
 
+	s.Inc.Lang = strings.Replace(strings.ToLower(s.Inc.Lang), "-", "_", 1)
+
 	if lowerSql == "select database()" {
 		return s.execute(ctx, sql)
 	} else if strings.HasPrefix(lowerSql, "select high_priority") {
@@ -6185,6 +6187,9 @@ func (s *session) executeInceptionSet(node *ast.InceptionSetStmt, sql string) ([
 			if err != nil {
 				return nil, err
 			}
+			if prefix == "lang" {
+				s.Inc.Lang = strings.Replace(strings.ToLower(s.Inc.Lang), "-", "_", 1)
+			}
 		}
 	}