fix nil handling for legacy servers

apache · Apr 13, 2024 · c4aab03 · c4aab03
1 parent 64088a2
commit c4aab03
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/go/arrow/flight/flightsql/client.go b/go/arrow/flight/flightsql/client.go
@@ -1236,10 +1236,17 @@ func (p *PreparedStatement) captureDoPutPreparedStatementHandle(pstream pb.Fligh
  if result, err = pstream.Recv(); err != nil && err != io.EOF {
  return err
  }
+ // skip if server does not provide a response (legacy server)
+ if result == nil {
+ return nil
+ }
  if err = proto.Unmarshal(result.GetAppMetadata(), &preparedStatementResult); err != nil {
  return err
  }
- p.handle = preparedStatementResult.GetPreparedStatementHandle()
+ handle := preparedStatementResult.GetPreparedStatementHandle()
+ if handle != nil {
+ p.handle = handle
+ }
  return nil
 }