change refcount, avoid ShmMsgQueue AddRef on send.

lichao

2021-05-14 c0d61292ef6bdd8e2e93525b5df6a8ec5c017c32

change refcount, avoid ShmMsgQueue AddRef on send.

 box/center.cpp

@@ -270,8 +270,7 @@
    bool SendAllocMsg(ShmSocket &socket, const Address dest, const MsgI &msg)
    {
        RecordMsg(msg);
        auto onExpireFree = [this, msg](const SendQ::Data &) { msgs_.FreeMsg(msg.id()); };
        return socket.Send(dest, msg, onExpireFree);
        return socket.Send(dest, msg);
    }

    void OnAlloc(ShmSocket &socket, const int64_t val)

 src/sendq.cpp

@@ -40,26 +40,8 @@
    }

    auto SendData = [&](Data &d) {
        auto TryLoop = [&](auto &&data) {
            for (int i = 0; i < 1; ++i) {
                if (mq.TrySend(remote, data)) {
                    return true;
                }
            }
            return false;
        };
        bool r = false;
        if (d.index() == 0) {
            auto &msg = boost::variant2::get<0>(pos->data().data_);
            r = TryLoop(msg);
            if (r) {
                msg.Release();
            }
        } else {
            auto command = boost::variant2::get<1>(pos->data().data_);
            r = TryLoop(command);
        }
        return r;
        auto TrySend = [&](auto &&data) { return mq.TrySend(remote, data); };
        return boost::variant2::visit(TrySend, pos->data().data_);
    };

    while (pos != arr.end() && SendData(pos->data().data_)) {

 src/shm_msg_queue.h

@@ -75,14 +75,7 @@
    bool TryRecv(MsgI &msg) { return TryRecv(msg.OffsetRef()); }
    static Queue *Find(ShmType &shm, const MQId remote_id);
    static bool TrySend(ShmType &shm, const MQId remote_id, const RawData val);
    static bool TrySend(ShmType &shm, const MQId remote_id, MsgI msg)
    {
        bool r = false;
        msg.AddRef(); // TODO check if we could avoid addref here.
        DEFER1(if (!r) { msg.Release(); });
        r = TrySend(shm, remote_id, msg.Offset());
        return r;
    }
    static bool TrySend(ShmType &shm, const MQId remote_id, MsgI msg) { return TrySend(shm, remote_id, msg.Offset()); }
    bool TrySend(const MQId remote_id, const MsgI &msg) { return TrySend(shm(), remote_id, msg); }
    bool TrySend(const MQId remote_id, const RawData val) { return TrySend(shm(), remote_id, val); }