check type assertion

yugabyte · Dec 27, 2024 · 65c787e · 65c787e
1 parent 69c7a4c
commit 65c787e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/yb-voyager/src/query/queryparser/helpers_protomsg.go b/yb-voyager/src/query/queryparser/helpers_protomsg.go
@@ -409,7 +409,10 @@ func GetSchemaAndObjectName(nameList protoreflect.List) (string, string) {
 }
 
 func ProtoAsSelectStmt(msg protoreflect.Message) (*pg_query.SelectStmt, error) {
-	protoMsg := msg.Interface().(proto.Message)
+	protoMsg, ok := msg.Interface().(proto.Message)
+	if !ok {
+		return nil, fmt.Errorf("failed to cast msg to proto.Message")
+	}
 	selectStmtNode, ok := protoMsg.(*pg_query.SelectStmt)
 	if !ok {
 		return nil, fmt.Errorf("failed to cast msg to %s", PG_QUERY_SELECTSTMT_NODE)