fujuntang
2021-09-11 dc01e4cbb01e96d19b470a366bbe648d426ed171
src/socket/shm_mod_socket.h
@@ -3,15 +3,16 @@
#include "usg_common.h"
#include "shm_socket.h"
#include "shm_allocator.h"
#include "mem_pool.h"
#include "shm_mm.h"
#include "hashtable.h"
#include "proc_def.h"
#include "sem_util.h"
#include "logger_factory.h"
#include "key_def.h"
#include <set>
#include "socket_def.h"
#define BUS_HEAD_SIZE (64 + 2 * sizeof(uint32_t))
#define BUS_HEAD_SIZE sizeof(bus_head_t)
class BusServerSocket;
struct bus_head_t
@@ -36,7 +37,8 @@
   static int get_bus_sendbuf(bus_head_t &request_head, const void *topic_buf, int topic_size, const void *content_buf, int content_size, void **retbuf);
public:
   static size_t remove_keys(int keys[], size_t length);
   // static size_t remove_keys(int keys[], size_t length);
   // static size_t remove_keys_exclude(int keys[], size_t length);
  // bus header 编码为网络传输的字节
  static void * encode_bus_head(bus_head_t & bushead);
@@ -47,6 +49,7 @@
   ShmModSocket();
   ~ShmModSocket();
    
   int stop();
   /**
    * 绑定端口到socket, 如果不绑定则系统自动分配一个
    * @return 0 成功, 其他值 失败的错误码
@@ -58,23 +61,13 @@
    * @return 0 成功, 其他值 失败的错误码
   */
   int force_bind(int key);
   /**
    * 发送信息
    * @key 发送给谁
    * @flag BUS_TIMEOUT_FLAG  BUS_NOWAIT_FLAG
    * @return 0 成功, 其他值 失败的错误码
    */
   int sendto(const void *buf, const int size, const int key, const struct timespec *timeout = NULL, int flag = 0);
   /**
    * 接收信息
    * @key 从谁哪里收到的信息
    * @return 0 成功, 其他值 失败的错误码
   */
   int recvfrom(void **buf, int *size, int *key,  const struct timespec *timeout = NULL, int flag = 0);
  int bind_proc_id(char *buf, int len);
  int reg(void *pData, int len, void **buf, int *size, const int timeout_ms, int flag);
  int sendto(const void *buf, const int size, const int key, const struct timespec *timeout = NULL, int flag = 0, int reset = 0, int data_set = 0);
  int recvfrom(void **buf, int *size, int *key,  const struct timespec *timeout = NULL, int flag = 0, int reset = 0, int data_set = 0);
   /**
    * 发送请求信息并等待接收应答
@@ -87,9 +80,10 @@
    const struct timespec *timeout = NULL, int flag = 0);
   // 超时返回。 @sec 秒 , @nsec 纳秒
   int sendandrecv_unsafe(const void *send_buf, const int send_size, const int key, void **recv_buf, int *recv_size,
    const  struct timespec *timeout = NULL, int flag = 0) ;
   /**
    *
    */
  int recvandsend( recvandsend_callback_fn callback, const struct timespec *timeout = NULL , int flag = 0, void * user_data = NULL);
   /**
    * 订阅指定主题
@@ -125,7 +119,14 @@
    */
   int get_key() ;
  int get_procid(char *buf, int len);
};
typedef std::map<int, ProcInfo, std::less<int>, SHM_STL_Allocator<std::pair<int, ProcInfo> > > ProcZone;
typedef std::set<SHMString,  std::less<SHMString>, SHM_STL_Allocator<SHMString> > TcsZone;
typedef std::map<int, TcsZone *, std::less<int>, SHM_STL_Allocator<std::pair<const int, TcsZone *> > > ProcTcsMap;
#endif