Add topics sub and request support.

fujuntang

2021-09-11 dc01e4cbb01e96d19b470a366bbe648d426ed171

 src/socket/shm_mod_socket.h

@@ -3,161 +3,130 @@
#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 ACTION_LIDENTIFIER "<**"
#define ACTION_RIDENTIFIER "**>"
#define TOPIC_LIDENTIFIER "{"
#define TOPIC_RIDENTIFIER "}"
#define BUS_HEAD_SIZE sizeof(bus_head_t)
class BusServerSocket;

static Logger *logger = LoggerFactory::getLogger();
#define BUS_MAP_KEY 1
//typedef std::basic_string<char, std::char_traits<char>, SHM_STL_Allocator<char> > SHMString;
typedef std::set<int,  std::less<int>, SHM_STL_Allocator<int> > SHMKeySet;
typedef std::map<SHMString, SHMKeySet *, std::less<SHMString>, SHM_STL_Allocator<std::pair<const SHMString, SHMKeySet *> > > SHMTopicSubMap;

enum socket_mod_t
struct bus_head_t
{
   PULL_PUSH = 1,
   REQ_REP = 2,
   PAIR = 3,
   PUB_SUB = 4,
   SURVEY = 5,
   BUS = 6
	
   char action[64];
   uint32_t topic_size;
   uint32_t content_size;
};


class ShmModSocket {
friend class BusServerSocket;
private:
   shm_socket_t *shm_socket;
  socket_mod_t mod;
  // pthread_t recv_thread;
  // <主题， 订阅者>
   SHMTopicSubMap *topic_sub_map;
  
   std::set<int> *bus_set;

private:
   inline int _recvfrom_(void **buf, int *size, int *port,  struct timespec *timeout, int flags);
   void _proxy_sub( char *topic, int port);
   void _proxy_pub( char *topic, size_t head_len, void *buf, size_t size, int port);
   void *run_pubsub_proxy();
   int parse_pubsub_topic(char *str, size_t size, char **_action, char **_topic, size_t *head_len );
   int _sub_( char *topic, int size, int port, struct timespec *timeout, int flags);
   int _pub_( char *topic, int topic_size, void *content, int content_size, int port, struct timespec *timeout, int flags);
	 
	 

   void _proxy_desub( char *topic, int port);
   void _proxy_desub_all(int port);
   int  _desub_( char *topic, int size, int port, struct timespec *timeout, int flags);
   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);

   static void foreach_subscripters(std::function<void(SHMKeySet *, int)>  cb);
   static bool include_in_keys(int key, int keys[], size_t length);
   static size_t remove_subscripters(int keys[], size_t length) ;
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);
  // 解码 bus  header
  static bus_head_t  decode_bus_head(void *headbs); 
  
public:
   ShmModSocket();
   ~ShmModSocket();
    
   int stop();
   /**
    * 绑定端口到socket, 如果不绑定则系统自动分配一个
    * @return 0 成功， 其他值 失败的错误码
   */
   int bind(int port);
   int bind(int key);

   /**
    * 强制绑定端口到socket, 适用于程序非正常关闭的情况下，重启程序绑定原来还没释放的key
    * @return 0 成功， 其他值 失败的错误码
   */
   int force_bind(int port);
   /**
    * 发送信息
    * @port 发送给谁
    * @return 0 成功， 其他值 失败的错误码
    */
   int sendto(const void *buf, const int size, const int port);
   // 发送信息超时返回。 @sec 秒 ， @nsec 纳秒
   int sendto_timeout(const void *buf, const int size, const int port, const struct timespec *timeout);
   // 发送信息立刻返回。
   int sendto_nowait(const void *buf, const int size, const int port);
   int force_bind(int key);

   /**
    * 接收信息
    * @port 从谁哪里收到的信息
    * @return 0 成功， 其他值 失败的错误码
   */
   int recvfrom(void **buf, int *size, int *port);
   // 接受信息超时返回。 @sec 秒 ， @nsec 纳秒
   int recvfrom_timeout(void **buf, int *size, int *port,  struct timespec *timeout);
   int recvfrom_nowait(void **buf, int *size, int *port);
  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);

   /**
    * 发送请求信息并等待接收应答
    * @port 发送给谁
    * @key 发送给谁
    * @flag BUS_TIMEOUT_FLAG  BUS_NOWAIT_FLAG
    * @return 0 成功， 其他值 失败的错误码
   */
   int sendandrecv(const void *send_buf, const int send_size, const int port, void **recv_buf, int *recv_size) ;
   // 超时返回。 @sec 秒 ， @nsec 纳秒
   int sendandrecv_timeout(const void *send_buf, const int send_size, const int port, void **recv_buf, int *recv_size,  struct timespec *timeout) ;
   int sendandrecv_nowait(const void *send_buf, const int send_size, const int port, void **recv_buf, int *recv_size) ;

   int sendandrecv_unsafe(const void *send_buf, const int send_size, const int port, void **recv_buf, int *recv_size) ;
   // 超时返回。 @sec 秒 ， @nsec 纳秒
   int sendandrecv_unsafe_timeout(const void *send_buf, const int send_size, const int port, void **recv_buf, int *recv_size,  struct timespec *timeout) ;
   int sendandrecv_unsafe_nowait(const void *send_buf, const int send_size, const int port, void **recv_buf, int *recv_size) ;
 
   int sendandrecv(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);


   /**
    * 启动bus
    * 
    * @return 0 成功， 其他值 失败的错误码
   */
   int  start_bus();
    */
  int recvandsend( recvandsend_callback_fn callback, const struct timespec *timeout = NULL , int flag = 0, void * user_data = NULL);

   /**
    * 订阅指定主题
    * @topic 主题
    * @size 主题长度
    * @port 总线端口
    * @key 总线端口
    * @flag BUS_TIMEOUT_FLAG  BUS_NOWAIT_FLAG
    */
   int  sub(char *topic, int size, int port);
   // 超时返回。 @sec 秒 ， @nsec 纳秒
   int  sub_timeout(char *topic, int size, int port,  struct timespec *timeout);
   int  sub_nowait(char *topic, int size, int port);
   int  sub(const char *topic, int size, int key,  const struct timespec *timeout = NULL, int flag = 0);


    /**
    * 取消订阅指定主题
     * @topic 主题,主题为空时取消全部订阅
    * @size 主题长度
    * @port 总线端口
    * @key 总线端口
    * @flag BUS_TIMEOUT_FLAG  BUS_NOWAIT_FLAG
    */
   int desub( char *topic, int size, int port);
   // 超时返回。 @sec 秒 ， @nsec 纳秒
   int desub_timeout(char *topic, int size, int port, struct timespec *timeout);
   int desub_nowait(char *topic, int size, int port) ;
   int desub(const char *topic, int size, int key, const struct timespec *timeout = NULL, int flag = 0);

   /**
    * 发布主题
    * @topic 主题
    * @content 主题内容
    * @port 总线端口
    * @key 总线端口
    * @flag BUS_TIMEOUT_FLAG  BUS_NOWAIT_FLAG
    */
   int  pub(char *topic, int topic_size, void *content, int content_size, int port);
   //  超时返回。 @sec 秒 ， @nsec 纳秒
   int  pub_timeout(char *topic, int topic_size, void *content, int content_size, int port,  struct timespec *timeout);
   int  pub_nowait(char *topic, int topic_size, void *content, int content_size, int port);
   int  pub(const char *topic, int topic_size, const void *content, int content_size, int key, const  struct timespec *timeout = NULL, int flag = 0);


   /**
    * 获取soket端口号
    * 获取soket key
    */
   int get_port() ;
   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