NameServiceOnTriples.h

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: 2006-2021 Istituto Italiano di Tecnologia (IIT)
 * SPDX-FileCopyrightText: 2006-2010 RobotCub Consortium
 * SPDX-License-Identifier: BSD-3-Clause
 */
 
#ifndef YARP_SERVERSQL_IMPL_NAMESERVICEONTRIPLES_H
#define YARP_SERVERSQL_IMPL_NAMESERVICEONTRIPLES_H
 
#include <yarp/name/NameService.h>
#include <yarp/serversql/impl/TripleSource.h>
#include <yarp/serversql/impl/Allocator.h>
#include <yarp/serversql/impl/Subscriber.h>
#include <yarp/os/NameStore.h>
#include <yarp/os/Semaphore.h>
 
#include <mutex>
 
namespace yarp {
namespace serversql {
namespace impl {
 
class NameTripleState
{
public:
    yarp::os::Bottle& cmd;
    yarp::os::Bottle& reply;
    yarp::os::Bottle& event;
    const yarp::os::Contact& remote;
    TripleSource& mem;
    bool bottleMode;
    bool nestedMode;
 
    NameTripleState(yarp::os::Bottle& cmd,
                    yarp::os::Bottle& reply,
                    yarp::os::Bottle& event,
                    const yarp::os::Contact& remote,
                    TripleSource& mem) :
            cmd(cmd),
            reply(reply),
            event(event),
            remote(remote),
            mem(mem),
            bottleMode(false),
            nestedMode(false)
    {
    }
};
 
class NameServiceOnTriples : public yarp::name::NameService
{
private:
    TripleSource *db;
    Allocator *alloc;
    Subscriber *subscriber;
    std::string lastRegister;
    yarp::os::Contact serverContact;
    std::mutex mutex;
    yarp::os::Semaphore access;
    bool gonePublic;
    bool silent;
    yarp::os::NameSpace *delegate;
public:
    NameServiceOnTriples() :
            db(nullptr),
            alloc(nullptr),
            subscriber(nullptr),
            lastRegister(""),
            mutex(),
            access(1),
            gonePublic(false),
            silent(false),
            delegate(nullptr)
    {
    }
 
    void open(TripleSource *db,
              Allocator *alloc,
              const yarp::os::Contact& serverContact)
    {
        this->db = db;
        this->alloc = alloc;
        this->serverContact = serverContact;
    }
 
    void setSubscriber(Subscriber *subscriber)
    {
        this->subscriber = subscriber;
    }
 
    void setSilent(bool flag)
    {
        this->silent = flag;
    }
 
    yarp::os::Contact query(const std::string& portName,
                            NameTripleState& act,
                            const std::string& prefix,
                            bool nested = false);
 
    bool announce(const std::string& name, int activity) override;
 
    yarp::os::Contact query(const std::string& portName) override;
 
    bool cmdQuery(NameTripleState& act, bool nested = false);
 
    bool cmdRegister(NameTripleState& act);
 
    bool cmdUnregister(NameTripleState& act);
 
    bool cmdList(NameTripleState& act);
 
    bool cmdListRunners(NameTripleState& act);
 
    bool cmdSet(NameTripleState& act);
 
    bool cmdGet(NameTripleState& act);
 
    bool cmdCheck(NameTripleState& act);
 
    bool cmdRoute(NameTripleState& act);
 
    bool cmdGc(NameTripleState& act);
 
    bool cmdHelp(NameTripleState& act);
 
    bool apply(yarp::os::Bottle& cmd,
               yarp::os::Bottle& reply,
               yarp::os::Bottle& event,
               const yarp::os::Contact& remote) override;
 
    void goPublic() override
    {
        gonePublic = true;
    }
 
    void lock() override;
 
    void unlock() override;
 
    void setDelegate(yarp::os::NameSpace *delegate)
    {
        this->delegate = delegate;
    }
 
    yarp::os::NameSpace *getDelegate()
    {
        return delegate;
    }
};
 
} // namespace impl
} // namespace serversql
} // namespace yarp
 
 
#endif // YARP_SERVERSQL_IMPL_NAMESERVICEONTRIPLES_H