improve Coerce to cast all columns with '*' [nt]

flarco · Nov 25, 2023 · 3c22382 · 3c22382
1 parent bc540fe
commit 3c22382
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/iop/datatype.go b/iop/datatype.go
@@ -386,6 +386,18 @@ func (cols Columns) Coerce(castCols Columns, hasHeader bool) (newCols Columns) {
 				newCols[i].Type = StringType
 			}
 		}
+
+		if len(castCols) == 1 && castCols[0].Name == "*" {
+			col = castCols[0]
+			if col.Type.IsValid() {
+				g.Debug("casting column '%s' as '%s'", newCols[i].Name, col.Type)
+				newCols[i].Type = col.Type
+			} else {
+				g.Warn("Provided unknown column type (%s) for column '%s'. Using string.", col.Type, col.Name)
+				newCols[i].Type = StringType
+			}
+
+		}
 	}
 	return newCols
 }