File shard_proxy_io.h¶

File List > broker > shard_proxy_io.h
/*
 * Copyright 2008 Search Solution Corporation
 * Copyright 2016 CUBRID Corporation
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 *
 */


/*
 * shard_proxy_io.h -
 */

#ifndef _SHARD_PROXY_IO_H_
#define _SHARD_PROXY_IO_H_

#ident "$Id$"

#include <fcntl.h>
#if !defined(WINDOWS)
#include <netdb.h>
#include <sys/socket.h>
#include <sys/un.h>
#include <sys/time.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#endif /* !WINDOWS */

#include "broker_recv_fd.h"
#include "cas_network.h"
#include "shard_proxy_common.h"
#include "shard_metadata.h"

#if defined(WINDOWS)
#define CLOSESOCKET(fd)                 closesocket((SOCKET)fd)
#define READSOCKET(fd, buf, len)        recv((SOCKET)fd, buf, len, 0)
#define WRITESOCKET(fd, buf, len)       send((SOCKET)fd, buf, len, 0)
#else
#define CLOSESOCKET(fd)                 close(fd)
#define READSOCKET(fd, buf, len)        read(fd, buf, (size_t)len)
#define WRITESOCKET(fd, buf, len)       write(fd, buf, (size_t)len)
#endif

#define DEFAULT_POLL_INTERVAL   1

/* connection protocol(size : information)
PROTOCOL_SIZE : CAS_CONNECTION_REPLY_SIZE
CAS_INFO_SIZE : CAS_INFO_SIZE
CAS_PID_SIZE : proxy pid
BROKER_INFO_SIZE : broker_info
CAS_INFO_SIZE : cas_info
*/

#define PROTOCOL_SIZE           sizeof(int)
#define PROXY_CONNECTION_REPLY_SIZE(con_reply_size)     (PROTOCOL_SIZE + CAS_INFO_SIZE + (con_reply_size))

#define SHARD_TEMPORARY_UNAVAILABLE     (-1)

#define PROXY_IO_FROM_CAS               (true)
#define PROXY_IO_FROM_CLIENT            (false)
#define PROXY_CONV_ERR_TO_NEW           (true)
#define PROXY_CONV_ERR_TO_OLD           (false)

#if !defined(LINUX)
/* for network global variables */
extern fd_set rset, wset, allset, wnewset, wallset;
extern int maxfd;
#endif /* !LINUX */

extern T_CLIENT_IO_GLOBAL proxy_Client_io;
extern T_SHARD_IO_GLOBAL proxy_Shard_io;

extern void set_data_length (char *buffer, int length);
extern int get_data_length (char *buffer);
extern int get_msg_length (char *buffer);

extern char *proxy_dup_msg (char *msg);

extern void proxy_set_con_status_in_tran (char *msg);
extern void proxy_set_con_status_out_tran (char *msg);
extern void proxy_set_force_out_tran (char *msg);
extern void proxy_unset_force_out_tran (char *msg);
extern int proxy_make_net_buf (T_NET_BUF * net_buf, int size, T_BROKER_VERSION client_version);

extern int proxy_io_make_error_msg (char *driver_info, char **buffer, int error_ind, int error_code,
                    const char *error_msg, char is_in_tran);
extern int proxy_io_make_no_error (char *driver_info, char **buffer);
extern int proxy_io_make_con_close_ok (char *driver_info, char **buffer);
extern int proxy_io_make_end_tran_ok (char *driver_info, char **buffer);
extern int proxy_io_make_check_cas_ok (char *driver_info, char **buffer);
extern int proxy_io_make_set_db_parameter_ok (char *driver_info, char **buffer);
extern int proxy_io_make_ex_get_isolation_level (char *driver_info, char **buffer, void *argv);
extern int proxy_io_make_ex_get_lock_timeout (char *driver_info, char **buffer, void *argv);

extern int proxy_io_make_end_tran_request (char *driver_info, char **buffer, bool commit);
#if defined (ENABLE_UNUSED_FUNCTION)
extern int proxy_io_make_end_tran_commit (char **buffer);
#endif /* ENABLE_UNUSED_FUNCTION */
extern int proxy_io_make_end_tran_abort (char *driver_info, char **buffer);
extern int proxy_io_make_close_req_handle_ok (char *driver_info, char **buffer, bool is_in_tran);
#if defined (ENABLE_UNUSED_FUNCTION)
extern int proxy_io_make_close_req_handle_in_tran_ok (char **buffer);
#endif /* ENABLE_UNUSED_FUNCTION */
extern int proxy_io_make_close_req_handle_out_tran_ok (char *driver_info, char **buffer);
extern int proxy_io_make_cursor_close_out_tran_ok (char *driver_info, char **buffer);

extern int proxy_io_make_get_db_version (char *driver_info, char **buffer);
extern int proxy_io_make_client_conn_ok (char *driver_info, char **buffer);
extern int proxy_io_make_client_proxy_alive (char *driver_info, char **buffer);
extern int proxy_io_make_client_dbinfo_ok (char *driver_info, char **buffer);
extern int proxy_io_make_client_acl_fail (char *driver_info, char **buffer);
extern int proxy_io_make_shard_info (char *driver_info, char **buffer);
extern int proxy_io_make_check_cas (char *driver_info, char **buffer);

extern int proxy_socket_set_write_event (T_SOCKET_IO * sock_io_p, T_PROXY_EVENT * event_p);
extern void proxy_io_buffer_clear (T_IO_BUFFER * io_buffer);
extern void proxy_socket_io_print (bool print_all);
extern void proxy_client_io_print (bool print_all);
extern char *proxy_str_client_io (T_CLIENT_IO * cli_io_p);

extern void proxy_shard_io_print (bool print_all);
extern char *proxy_str_cas_io (T_CAS_IO * cas_io_p);

extern void proxy_client_io_free (T_CLIENT_IO * cli_io_p);
extern void proxy_client_io_free_by_ctx (int client_id, int ctx_cid, int ctx_uid);
extern T_CLIENT_IO *proxy_client_io_find_by_ctx (int client_id, int ctx_cid, unsigned int ctx_uid);
extern T_CLIENT_IO *proxy_client_io_find_by_fd (int client_id, SOCKET fd);
extern int proxy_client_io_write (T_CLIENT_IO * cli_io_p, T_PROXY_EVENT * event_p);
extern void proxy_cas_io_free_by_ctx (int shard_id, int cas_id, int ctx_cid, unsigned int ctx_uid);
extern T_CAS_IO *proxy_cas_find_io_by_ctx (int shard_id, int cas_id, int ctx_cid, unsigned int ctx_uid);

extern T_CAS_IO *proxy_cas_alloc_by_ctx (int client_id, int shard_id, int cas_id, int ctx_cid, unsigned int ctx_uid,
                     int timeout, int func_code);
extern void proxy_cas_release_by_ctx (int shard_id, int cas_id, int ctx_cid, unsigned int ctx_uid);
extern int proxy_cas_io_write (T_CAS_IO * cas_io_p, T_PROXY_EVENT * event_p);

extern int proxy_io_close_all_fd (void);

extern int proxy_io_process (void);
extern int proxy_io_initialize (void);
extern void proxy_io_destroy (void);


extern int proxy_socket_io_delete (SOCKET fd);
extern int proxy_io_set_established_by_ctx (T_PROXY_CONTEXT * ctx_p);

extern char *proxy_get_driver_info_by_ctx (T_PROXY_CONTEXT * ctx_p);
extern char *proxy_get_driver_info_by_fd (T_SOCKET_IO * sock_io_p);

extern void proxy_available_cas_wait_timer (void);
extern int proxy_convert_error_code (int error_ind, int error_code, char *driver_info, T_BROKER_VERSION client_version,
                     bool to_new);

extern int proxy_context_direct_send_error (T_PROXY_CONTEXT * ctx_p);
#endif /* _SHARD_PROXY_IO_H_ */