134 lines
4.5 KiB
C
134 lines
4.5 KiB
C
/*
|
|
* Copyright (c) 2020 YuQing <384681@qq.com>
|
|
*
|
|
* This program is free software: you can use, redistribute, and/or modify
|
|
* it under the terms of the Lesser GNU General Public License, version 3
|
|
* or later ("LGPL"), as published by the Free Software Foundation.
|
|
*
|
|
* This program is distributed in the hope that it will be useful, but WITHOUT
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
* FITNESS FOR A PARTICULAR PURPOSE.
|
|
*
|
|
* You should have received a copy of the Lesser GNU General Public License
|
|
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
//multi_socket_client.h
|
|
|
|
#ifndef _MULTI_SOCKET_CLIENT_H_
|
|
#define _MULTI_SOCKET_CLIENT_H_
|
|
|
|
#include <net/if.h>
|
|
#include <string.h>
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/socket.h>
|
|
#include "common_define.h"
|
|
#include "connection_pool.h"
|
|
#include "fast_buffer.h"
|
|
#include "ioevent.h"
|
|
|
|
typedef enum {
|
|
fms_stage_recv_header = 'H',
|
|
fms_stage_recv_body = 'B'
|
|
} FastMultiSockRecvStage;
|
|
|
|
struct fast_multi_sock_client;
|
|
struct fast_multi_sock_entry;
|
|
|
|
typedef int64_t (*fms_client_get_current_time_ms_func)();
|
|
|
|
//return the body length
|
|
typedef int (*fms_client_get_body_length_func)(const FastBuffer *recv_buffer);
|
|
|
|
//IO deal fucntion
|
|
typedef int (*fast_multi_sock_client_io_func)(struct fast_multi_sock_client *
|
|
client, struct fast_multi_sock_entry *entry);
|
|
|
|
typedef struct fast_multi_sock_entry {
|
|
ConnectionInfo *conn; //the connected socket must be non-block socket
|
|
FastBuffer *send_buffer; //send buffer for internal use
|
|
fast_multi_sock_client_io_func io_callback; //for internal use
|
|
FastBuffer recv_buffer; //recv buffer for response package
|
|
int error_no; //0 for success, != 0 fail
|
|
int remain; //remain bytes, for internal use
|
|
FastMultiSockRecvStage recv_stage; //for internal use
|
|
bool done; //for internal use
|
|
} FastMultiSockEntry;
|
|
|
|
typedef struct fast_multi_sock_client {
|
|
int entry_count;
|
|
int header_length; //package header size
|
|
int pulling_count;
|
|
int success_count;
|
|
int timeout_ms;
|
|
int64_t deadline_time_ms;
|
|
FastMultiSockEntry *entries;
|
|
fms_client_get_current_time_ms_func get_current_time_ms_func;
|
|
fms_client_get_body_length_func get_body_length_func;
|
|
IOEventPoller ioevent;
|
|
} FastMultiSockClient;
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
|
|
/**
|
|
init function
|
|
@param client the client context
|
|
@param entries the socket entries
|
|
@param entry_count the count of socket entries
|
|
@param header_length the header length of a package
|
|
@param get_body_length_func the get body length function
|
|
@param get_current_time_ms_func the get current time in ms function
|
|
@param init_recv_buffer_size the initial size of response buffer
|
|
@param timeout_ms the timeout in milliseconds
|
|
@return error no, 0 for success, != 0 fail
|
|
*/
|
|
int fast_multi_sock_client_init_ex(FastMultiSockClient *client,
|
|
FastMultiSockEntry *entries, const int entry_count,
|
|
const int header_length,
|
|
fms_client_get_body_length_func get_body_length_func,
|
|
fms_client_get_current_time_ms_func get_current_time_ms_func,
|
|
const int init_recv_buffer_size, const int timeout_ms);
|
|
|
|
/**
|
|
init function
|
|
@param client the client context
|
|
@param entries the socket entries
|
|
@param entry_count the count of socket entries
|
|
@param header_length the header length of a package
|
|
@param get_body_length_func the get body length function
|
|
@param init_recv_buffer_size the initial size of response buffer
|
|
@param timeout the timeout in seconds
|
|
@return error no, 0 for success, != 0 fail
|
|
*/
|
|
int fast_multi_sock_client_init(FastMultiSockClient *client,
|
|
FastMultiSockEntry *entries, const int entry_count,
|
|
const int header_length,
|
|
fms_client_get_body_length_func get_body_length_func,
|
|
const int init_recv_buffer_size, const int timeout);
|
|
|
|
/**
|
|
destroy function
|
|
@param client the client context
|
|
@return none
|
|
*/
|
|
void fast_multi_sock_client_destroy(FastMultiSockClient *client);
|
|
|
|
/**
|
|
request function
|
|
@param client the client context
|
|
@param send_buffer the buffer to send
|
|
@return error no, 0 for success, != 0 fail
|
|
*/
|
|
int fast_multi_sock_client_request(FastMultiSockClient *client,
|
|
FastBuffer *send_buffer);
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
#endif
|