src/KingSystem/Utils/Thread/MessageReceiverEx.cpp


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

#include "KingSystem/Utils/Thread/MessageReceiverEx.h"

namespace ksys {

MessageReceiverEx::MessageReceiverEx() = default;

MessageReceiverEx::~MessageReceiverEx() = default;

int MessageReceiverEx::receive(const Message& message) {
    return handleMessage(message);
}

void MessageReceiverEx::receive(const MessageAck& ack) {
    handleAck(ack);
    --mNumPendingAcks;
    static_cast<void>(mNumPendingAcks.load());
}

bool MessageReceiverEx::checkFlag() const {
    return mFlag && *mFlag == 1;
}

bool MessageReceiverEx::isWaitingForAck() const {
    return mNumPendingAcks > 0;
}

void MessageReceiverEx::setFlag(bool ack_requested) {
    if (mFlag) {
        if (ack_requested)
            mNumPendingAcks.increment();
        *mFlag = 1;
    }
}

int MessageReceiverEx::handleMessage(const Message& message) {
    return true;
}

void MessageReceiverEx::handleAck(const MessageAck& ack) {}

void MessageReceiverEx::setFlagPointer(u8* ptr) {
    if (ptr) {
        mFlag = ptr;
        *mFlag = 0;
    }
}

void MessageReceiverEx::clearFlagPointer() {
    *mFlag = 0;
    mFlag = nullptr;
}

}  // namespace ksys