Merge pull request #1140 from fengzeroz/main

chore(connection): return the result of conn recv
This commit is contained in:
fengzeroz
2023-03-24 16:56:45 +08:00
committed by GitHub
2 changed files with 11 additions and 8 deletions
+4 -4
View File
@@ -266,11 +266,11 @@ ssize_t neu_conn_tcp_server_recv(neu_conn_t *conn, int fd, uint8_t *buf,
typedef int (*neu_conn_stream_consume_fn)(
void *context, neu_protocol_unpack_buf_t *protocol_buf);
void neu_conn_stream_consume(neu_conn_t *conn, void *context,
neu_conn_stream_consume_fn fn);
int neu_conn_stream_consume(neu_conn_t *conn, void *context,
neu_conn_stream_consume_fn fn);
void neu_conn_stream_tcp_server_consume(neu_conn_t *conn, int fd, void *context,
neu_conn_stream_consume_fn fn);
int neu_conn_stream_tcp_server_consume(neu_conn_t *conn, int fd, void *context,
neu_conn_stream_consume_fn fn);
typedef neu_buf_result_t (*neu_conn_process_msg)(
void *context, neu_protocol_unpack_buf_t *protocol_buf);
+7 -4
View File
@@ -867,8 +867,8 @@ static void conn_tcp_server_del_client(neu_conn_t *conn, int fd)
}
}
void neu_conn_stream_consume(neu_conn_t *conn, void *context,
neu_conn_stream_consume_fn fn)
int neu_conn_stream_consume(neu_conn_t *conn, void *context,
neu_conn_stream_consume_fn fn)
{
ssize_t ret = neu_conn_recv(conn, conn->buf + conn->offset,
conn->buf_size - conn->offset);
@@ -898,10 +898,12 @@ void neu_conn_stream_consume(neu_conn_t *conn, void *context,
conn->offset);
}
}
return ret;
}
void neu_conn_stream_tcp_server_consume(neu_conn_t *conn, int fd, void *context,
neu_conn_stream_consume_fn fn)
int neu_conn_stream_tcp_server_consume(neu_conn_t *conn, int fd, void *context,
neu_conn_stream_consume_fn fn)
{
ssize_t ret = neu_conn_tcp_server_recv(conn, fd, conn->buf + conn->offset,
conn->buf_size - conn->offset);
@@ -928,6 +930,7 @@ void neu_conn_stream_tcp_server_consume(neu_conn_t *conn, int fd, void *context,
conn->offset);
}
}
return ret;
}
int neu_conn_wait_msg(neu_conn_t *conn, void *context, uint16_t n_byte,