api/pybind_kv_service.cpp

#include <fcntl.h>
#include <getopt.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <unistd.h>
#include <pybind11/pybind11.h>

#include <fstream>
#include "common/proto/signature_info.pb.h"
#include "interface/kv/kv_client.h"
#include "platform/config/resdb_config_utils.h"

using resdb::GenerateReplicaInfo;
using resdb::GenerateResDBConfig;
using resdb::KVClient;
using resdb::ReplicaInfo;
using resdb::ResDBConfig;




std::string get(std::string key, std::string config_path) {
    ResDBConfig config = GenerateResDBConfig(config_path);
    config.SetClientTimeoutMs(100000);
    KVClient client(config);
    auto result_ptr = client.Get(key);
    if (result_ptr) {
        return *result_ptr;
    } else {
        return "";
    }
}

bool set(std::string key, std::string value, std::string config_path) {
    ResDBConfig config = GenerateResDBConfig(config_path);
    config.SetClientTimeoutMs(100000);
    KVClient client(config);
    int result = client.Set(key, value);
    if (result == 0) {
        return true;
    } else {
        return false;
    }
}

PYBIND11_MODULE(pybind_kv, m) {
    m.def("get", &get, "A function that gets a value from the key-value store");
    m.def("set", &set, "A function that sets a value in the key-value store");
}