runtime/vam: fix cast from string to string (#6281)

nwt · web-flow · commit 525853d2640d · 2025-10-06T14:33:24.000-04:00
diff --git a/runtime/vam/expr/cast/bytes.go b/runtime/vam/expr/cast/bytes.go
@@ -5,11 +5,15 @@ import (
 )
 
 func castToBytes(vec vector.Any, index []uint32) (vector.Any, []uint32, string, bool) {
-	strVec, ok := vec.(*vector.String)
-	if !ok {
+	var out vector.Any
+	switch vec := vec.(type) {
+	case *vector.Bytes:
+		out = vec
+	case *vector.String:
+		out = vector.NewBytes(vec.Table(), vector.NullsOf(vec))
+	default:
 		return nil, nil, "", false
 	}
-	out := vector.Any(vector.NewBytes(strVec.Table(), vector.NullsOf(strVec)))
 	if index != nil {
 		out = vector.Pick(out, index)
 	}
diff --git a/runtime/ztests/expr/cast/bytes.yaml b/runtime/ztests/expr/cast/bytes.yaml
@@ -3,13 +3,19 @@ spq: this::bytes
 vector: true
 
 input: |
+  0x01
+  null::bytes
+  null
   "foo"
   "bar"::=named
   ""
   null::string
   {x:"foo"}
 
 output: |
+  0x01
+  null::bytes
+  null::bytes
   0x666f6f
   0x626172
   0x

Original file line number	Diff line number	Diff line change
`@@ -5,11 +5,15 @@ import (`
`5`	`5`	`)`
`6`	`6`
`7`	`7`	`func castToBytes(vec vector.Any, index []uint32) (vector.Any, []uint32, string, bool) {`
`8`		`- strVec, ok := vec.(*vector.String)`
`9`		`- if !ok {`
	`8`	`+ var out vector.Any`
	`9`	`+ switch vec := vec.(type) {`
	`10`	`+ case *vector.Bytes:`
	`11`	`+ out = vec`
	`12`	`+ case *vector.String:`
	`13`	`+ out = vector.NewBytes(vec.Table(), vector.NullsOf(vec))`
	`14`	`+ default:`
`10`	`15`	`return nil, nil, "", false`
`11`	`16`	`}`
`12`		`- out := vector.Any(vector.NewBytes(strVec.Table(), vector.NullsOf(strVec)))`
`13`	`17`	`if index != nil {`
`14`	`18`	`out = vector.Pick(out, index)`
`15`	`19`	`}`