check before casting to string

PeerDB-io · Nov 14, 2023 · 6ff107f · 6ff107f
1 parent 13e0e0f
commit 6ff107f
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/flow/model/model.go b/flow/model/model.go
@@ -124,10 +124,19 @@ func (r *RecordItems) toMap() (map[string]interface{}, error) {
 		var err error
 		switch v.Kind {
 		case qvalue.QValueKindString, qvalue.QValueKindJSON:
-			if len(v.Value.(string)) > 15*1024*1024 {
-				jsonStruct[col] = ""
+			if v.Value == nil {
+				jsonStruct[col] = nil
 			} else {
-				jsonStruct[col] = v.Value
+				strVal, ok := v.Value.(string)
+				if !ok {
+					return nil, fmt.Errorf("expected string value for column %s for %T", col, v.Value)
+				}
+
+				if len(strVal) > 15*1024*1024 {
+					jsonStruct[col] = ""
+				} else {
+					jsonStruct[col] = strVal
+				}
 			}
 		case qvalue.QValueKindTimestamp, qvalue.QValueKindTimestampTZ, qvalue.QValueKindDate,
 			qvalue.QValueKindTime, qvalue.QValueKindTimeTZ: