zeroshade commented on code in PR #3109:
URL: https://github.com/apache/arrow-adbc/pull/3109#discussion_r2193014302


##########
go/adbc/sqldriver/driver.go:
##########
@@ -459,15 +458,63 @@ func arrFromVal(val any) arrow.Array {
                dt = arrow.PrimitiveTypes.Date64
                buffers[1] = 
memory.NewBufferBytes((*[8]byte)(unsafe.Pointer(&v))[:])
        case []byte:
-               dt = arrow.BinaryTypes.Binary
-               buffers[1] = 
memory.NewBufferBytes(arrow.Int32Traits.CastToBytes([]int32{0, int32(len(v))}))
+               if dt == nil || dt.ID() == arrow.BINARY {
+                       dt = arrow.BinaryTypes.Binary
+                       buffers[1] = 
memory.NewBufferBytes(arrow.Int32Traits.CastToBytes([]int32{0, int32(len(v))}))
+               } else if dt.ID() == arrow.LARGE_BINARY {
+                       dt = arrow.BinaryTypes.LargeBinary
+                       buffers[1] = 
memory.NewBufferBytes(arrow.Int64Traits.CastToBytes([]int64{0, int64(len(v))}))
+               }
                buffers = append(buffers, memory.NewBufferBytes(v))
        case string:
-               dt = arrow.BinaryTypes.String
-               buffers[1] = 
memory.NewBufferBytes(arrow.Int32Traits.CastToBytes([]int32{0, int32(len(v))}))
-
+               if dt == nil || dt.ID() == arrow.STRING {
+                       dt = arrow.BinaryTypes.String
+                       buffers[1] = 
memory.NewBufferBytes(arrow.Int32Traits.CastToBytes([]int32{0, int32(len(v))}))
+               } else if dt.ID() == arrow.LARGE_STRING {
+                       dt = arrow.BinaryTypes.LargeString
+                       buffers[1] = 
memory.NewBufferBytes(arrow.Int64Traits.CastToBytes([]int64{0, int64(len(v))}))
+               }
                buf := unsafe.Slice(unsafe.StringData(v), len(v))
                buffers = append(buffers, memory.NewBufferBytes(buf))
+       case arrow.Time32:
+               if dt == nil || dt.ID() != arrow.TIME32 {
+                       panic(errors.New("can only create array from 
arrow.Time32 with known type"))

Review Comment:
   We would only run into the issue when they try to execute with the passed in 
params, not when it is prepared. The issue is if they pass an `arrow.Time32` 
when the server didn't provide a parameter schema, we don't know what type this 
should be (because of the time unit).
   
   That said, it's easy enough to propagate the error upwards instead of 
panicking, which i'll do.



-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: github-unsubscr...@arrow.apache.org

For queries about this service, please contact Infrastructure at:
us...@infra.apache.org

Reply via email to