61 lines
1.2 KiB
C++
61 lines
1.2 KiB
C++
|
#ifndef __UT_ROBOT_LEASE_SERVER_HPP__
|
||
|
|
#define __UT_ROBOT_LEASE_SERVER_HPP__
|
||
|
|
|
||
|
|
#include <unitree/robot/server/server_base.hpp>
|
||
|
|
|
||
|
|
namespace unitree
|
||
|
|
{
|
||
|
|
namespace robot
|
||
|
|
{
|
||
|
|
class LeaseCache
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
LeaseCache();
|
||
|
|
~LeaseCache();
|
||
|
|
|
||
|
|
void Set(int64_t id, const std::string& mName, int64_t lastModified = 0);
|
||
|
|
void Renewal(int64_t lastModified = 0);
|
||
|
|
void Clear();
|
||
|
|
|
||
|
|
int64_t GetLastModified() const;
|
||
|
|
int64_t GetId() const;
|
||
|
|
const std::string& GetName() const;
|
||
|
|
|
||
|
|
private:
|
||
|
|
int64_t mLastModified;
|
||
|
|
int64_t mId;
|
||
|
|
std::string mName;
|
||
|
|
};
|
||
|
|
|
||
|
|
class LeaseServer : public ServerBase
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
explicit LeaseServer(const std::string& name, int64_t term);
|
||
|
|
~LeaseServer();
|
||
|
|
|
||
|
|
void Init();
|
||
|
|
|
||
|
|
bool CheckRequestLeaseDenied(int64_t leaseId);
|
||
|
|
|
||
|
|
private:
|
||
|
|
void ServerRequestHandler(const RequestPtr& request);
|
||
|
|
|
||
|
|
int32_t Apply(const std::string& parameter, std::string& data);
|
||
|
|
int32_t Renewal(int64_t leaseId);
|
||
|
|
|
||
|
|
int64_t GenerateId(const std::string& name);
|
||
|
|
|
||
|
|
private:
|
||
|
|
int64_t mTerm;
|
||
|
|
LeaseCache mCache;
|
||
|
|
common::Mutex mMutex;
|
||
|
|
ServerStubPtr mServerStubPtr;
|
||
|
|
};
|
||
|
|
|
||
|
|
using LeaseServerPtr = std::shared_ptr<LeaseServer>;
|
||
|
|
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#endif//__UT_ROBOT_LEASE_SERVER_HPP__
|