fix silly bug: correctly remove wrapper

timeplus-io · Oct 24, 2024 · 2294238 · 2294238
1 parent d09aae5
commit 2294238
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/core/dbio/database/database_proton.go b/core/dbio/database/database_proton.go
@@ -328,10 +328,14 @@ func (conn *ProtonConn) processBatch(tableFName string, table Table, batch *iop.
 
 	for i, col := range batch.Columns {
 		dbType := strings.ToLower(col.DbType)
-		dbType = strings.TrimPrefix(dbType, "low_cardinality(")
-		dbType = strings.TrimPrefix(dbType, "nullable(")
-
-		dbType = strings.TrimSuffix(dbType, ")")
+		if strings.HasPrefix(dbType, "nullable(") {
+			dbType = strings.TrimPrefix(dbType, "nullable(")
+			dbType = strings.TrimSuffix(dbType, ")")
+		}
+		if strings.HasPrefix(dbType, "low_cardinality(") {
+			dbType = strings.TrimPrefix(dbType, "low_cardinality(")
+			dbType = strings.TrimSuffix(dbType, ")")
+		}
 
 		switch dbType {
 		case "int8":