Function formatSQLDatabaseNameWithRegex

datasource/postgresql/postgresql.go:290–296

formatSQLDatabaseNameWithRegex 只对 dbname.scheme.tabname 格式进行数据库名称格式化，转为 "dbname".scheme.tabname 在pgsql中，大小写是通过"" 双引号括起来区分的,默认pg都是转为小写的，所以这里转为 "dbname".scheme."tabname"

(sql string)

// formatSQLDatabaseNameWithRegex 只对 dbname.scheme.tabname 格式进行数据库名称格式化，转为 "dbname".scheme.tabname
// 在pgsql中，大小写是通过"" 双引号括起来区分的,默认pg都是转为小写的，所以这里转为 "dbname".scheme."tabname"
func formatSQLDatabaseNameWithRegex(sql string) string {
	// 匹配 from dbname.scheme.table_name 的模式
	// 使用捕获组来精确匹配数据库名称，确保后面跟着scheme和table
	re := regexp.MustCompile(`(?i)\bfrom\s+([a-zA-Z_][a-zA-Z0-9_]*)\s*\.\s*([a-zA-Z_][a-zA-Z0-9_]*)\s*\.\s*([a-zA-Z_][a-zA-Z0-9_]*)`)

	return re.ReplaceAllString(sql, `from "$1"."$2"."$3"`)
}

func extractColumns(sql string) ([]string, error) {
	// 将 SQL 转换为小写以简化匹配