lichao
2021-05-21 11f6c600e55ca5677f93624efe44d2605cdd908d
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
/*
 * =====================================================================================
 *
 *       Filename:  shm_msg_queue.h
 *
 *    Description:  
 *
 *        Version:  1.0
 *        Created:  2021年04月26日 16时25分21秒
 *       Revision:  none
 *       Compiler:  gcc
 *
 *         Author:  Li Chao (), lichao@aiotlink.com
 *   Organization:  
 *
 * =====================================================================================
 */
#ifndef SHM_MSG_QUEUE_D847TQXH
#define SHM_MSG_QUEUE_D847TQXH
 
#include "defs.h"
#include "msg.h"
#include "shm_queue.h"
 
using namespace bhome_shm;
using namespace bhome_msg;
 
#define BH_USE_ATOMIC_Q
 
class ShmMsgQueue
{
public:
    typedef int64_t RawData;
 
#ifdef BH_USE_ATOMIC_Q
    typedef ShmObject<SharedQ63<0>> Shmq;
#else
    typedef ShmObject<SharedQueue<RawData>> Shmq;
    // typedef robust::FMutex Mutex;
    // typedef robust::SemMutex Mutex;
    typedef robust::NullMutex Mutex;
    typedef robust::Guard<Mutex> Guard;
#endif
 
    typedef Shmq::Data Queue;
    typedef Shmq::ShmType ShmType;
    typedef uint64_t MQId;
 
    ShmMsgQueue(ShmType &segment, const MQId id, const int len);
    ShmMsgQueue(ShmType &segment, const bool create_or_else_find, const MQId id, const int len);
    ShmMsgQueue(const int64_t abs_addr, ShmType &segment, const MQId id);
    ~ShmMsgQueue();
    static bool Remove(ShmType &shm, const MQId id);
    MQId Id() const { return id_; }
    ShmType &shm() const { return queue_.shm(); }
    int64_t AbsAddr() const { return queue_.offset(); }
 
    bool Recv(RawData &val, const int timeout_ms)
    {
#ifndef BH_USE_ATOMIC_Q
        Guard lock(GetMutex(Id()));
#endif
        return queue().Read(val, timeout_ms);
    }
 
    bool TryRecv(RawData &val)
    {
#ifndef BH_USE_ATOMIC_Q
        Guard lock(GetMutex(Id()));
#endif
        return queue().TryRead(val);
    }
 
    bool Recv(MsgI &msg, const int timeout_ms) { return Recv(msg.OffsetRef(), timeout_ms); }
    bool TryRecv(MsgI &msg) { return TryRecv(msg.OffsetRef()); }
    static Queue *Find(ShmType &shm, const MQId remote);
    static bool TrySend(ShmType &shm, const MQInfo &remote, const RawData val);
    bool TrySend(const MQInfo &remote, const RawData val) { return TrySend(shm(), remote, val); }
 
private:
#ifndef BH_USE_ATOMIC_Q
    static Mutex &GetMutex(const MQId id);
#endif
    MQId id_;
    Queue &queue() { return *queue_.data(); }
    Shmq queue_;
};
 
#endif // end of include guard: SHM_MSG_QUEUE_D847TQXH