From e4591df89d568b9f23814249e253753cb8344abc Mon Sep 17 00:00:00 2001
From: wangha <wanghamax@gmail.com>
Date: Fri, 24 Jan 2025 04:05:12 -0500
Subject: [PATCH] * NEW [parquet] Fix the null ptr in parquet->payload_arr.

Signed-off-by: wangha <wanghamax@gmail.com>
---
 src/mqtt/protocol/exchange/exchange_server.c | 6 +++++-
 src/supplemental/nanolib/parquet/parquet.cc  | 2 ++
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/mqtt/protocol/exchange/exchange_server.c b/src/mqtt/protocol/exchange/exchange_server.c
index 259a50380..a2ce2cb3f 100644
--- a/src/mqtt/protocol/exchange/exchange_server.c
+++ b/src/mqtt/protocol/exchange/exchange_server.c
@@ -470,7 +470,10 @@ static void query_send_async(exchange_sock_t *s, struct cmd_data *cmd_data)
 							nng_msg *newmsg = NULL;
 							nng_msg_alloc(&newmsg, 0);
 							nni_msg_append(newmsg, parquet_decoded_data->data, parquet_decoded_data->len);
-							nng_sendmsg(*(s->pair0_sock), newmsg, 0);
+							if (s->pair0_sock)
+								nng_sendmsg(*(s->pair0_sock), newmsg, 0);
+							else
+								log_error("pair0_sock is null!!!!!!!!!");
 							/* NOTE: sleep 1000ms */
 							nng_msleep(1000);
 							stream_decoded_data_free(parquet_decoded_data);
@@ -630,6 +633,7 @@ exchange_sock_init(void *arg, nni_sock *sock)
 	nni_mtx_init(&s->mtx);
 	nni_id_map_init(&s->rbmsgmap, 0, 0, true);
 	s->isBusy = false;
+	s->pair0_sock = NULL;
 
 	nni_lmq_init(&s->lmq, 256);
 
diff --git a/src/supplemental/nanolib/parquet/parquet.cc b/src/supplemental/nanolib/parquet/parquet.cc
index 16553e021..37e8c14a5 100644
--- a/src/supplemental/nanolib/parquet/parquet.cc
+++ b/src/supplemental/nanolib/parquet/parquet.cc
@@ -1302,6 +1302,8 @@ static parquet_data_ret *parquet_read_payload(shared_ptr<parquet::RowGroupReader
         parquet_data_packet **payload_arr = read_column_data(col.reader, index_vector, batch_size, total_values_read);
         if (payload_arr) {
             ret_rows_vec.push_back(payload_arr);
+        } else {
+            ret_rows_vec.push_back(nullptr);
         }
     }