From 937fd4bb86b0a3934154b5a3b261ae707b26c1ad Mon Sep 17 00:00:00 2001 From: fengzero Date: Thu, 28 Sep 2023 07:45:50 +0000 Subject: [PATCH] fix(connection): server consume --- src/connection/connection.c | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/src/connection/connection.c b/src/connection/connection.c index 1dae9b64..73639d43 100644 --- a/src/connection/connection.c +++ b/src/connection/connection.c @@ -1295,15 +1295,13 @@ int neu_conn_stream_tcp_server_consume(neu_conn_t *conn, int fd, void *context, } else if (used == -1) { neu_conn_tcp_server_close_client(conn, fd); break; + } else { + conn->offset -= used; + memmove(conn->buf, conn->buf + used, conn->offset); + neu_protocol_unpack_buf_init(&protocol_buf, conn->buf, + conn->offset); } } - if (conn->offset != 0) { - conn->offset -= neu_protocol_unpack_buf_used_size(&protocol_buf); - memmove(conn->buf, - conn->buf + - neu_protocol_unpack_buf_used_size(&protocol_buf), - conn->offset); - } } return ret; }