mirror of
https://gitee.com/acl-dev/acl.git
synced 2024-12-13 16:35:35 +08:00
109 lines
2.8 KiB
C++
109 lines
2.8 KiB
C++
//
|
|
// Created by shuxin ¡¡¡¡zheng on 2023/7/31.
|
|
//
|
|
|
|
#include "stdafx.h"
|
|
#include "proto/redis_coder.h"
|
|
#include "proto/redis_object.h"
|
|
#include "redis_string.h"
|
|
|
|
namespace pkv {
|
|
|
|
redis_string::redis_string(shared_db& db, const redis_object &obj,
|
|
redis_coder& base)
|
|
: db_(db), obj_(obj), base_(base) {}
|
|
|
|
bool redis_string::set(redis_coder& result) {
|
|
auto& objs = obj_.get_objects();
|
|
if (objs.size() < 3) {
|
|
logger_error("invalid SET params' size=%zd", objs.size());
|
|
return false;
|
|
}
|
|
|
|
auto key = objs[1]->get_str();
|
|
if (key == nullptr || *key == 0) {
|
|
logger_error("key null");
|
|
return false;
|
|
}
|
|
|
|
auto value = objs[2]->get_str();
|
|
if (value == nullptr || *value == 0) {
|
|
logger_error("value null");
|
|
return false;
|
|
}
|
|
|
|
if (!var_cfg_disable_serialize) {
|
|
std::string buff;
|
|
redis_coder coder(base_.get_cache());
|
|
coder.create_object().create_child().set_string("string", true) // type
|
|
.create_child().set_number(-1); // expire time
|
|
coder.create_object().set_string(value);
|
|
coder.to_string(buff);
|
|
coder.clear();
|
|
|
|
//printf("buf=[%s]\n", buff.c_str());
|
|
|
|
if (!var_cfg_disable_save) {
|
|
if (!db_->set(key, buff.c_str())) {
|
|
logger_error("db set error, key=%s", key);
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
|
|
result.create_object().set_status("OK");
|
|
return true;
|
|
}
|
|
|
|
bool redis_string::get(redis_coder& result) {
|
|
auto& objs = obj_.get_objects();
|
|
if (objs.size() < 2) {
|
|
logger_error("invalid GET params' size=%zd", objs.size());
|
|
return false;
|
|
}
|
|
|
|
auto key = objs[1]->get_str();
|
|
if (key == nullptr || *key == 0) {
|
|
logger_error("key null");
|
|
return false;
|
|
}
|
|
|
|
std::string buff;
|
|
if (!db_->get(key, buff) || buff.empty()) {
|
|
logger_error("db get error, key=%s", key);
|
|
return false;
|
|
}
|
|
|
|
//printf(">>>get key=%s, val=[%s]\n", key, buff.c_str());
|
|
redis_coder builder(base_.get_cache());
|
|
size_t len = buff.size();
|
|
(void) builder.update(buff.c_str(), len);
|
|
if (len > 0) {
|
|
logger_error("invalid buff in db, key=%s", key);
|
|
return false;
|
|
}
|
|
|
|
auto& objs2 = builder.get_objects();
|
|
if (objs2.size() != 2) {
|
|
logger_error("invalid object in db, key=%s, size=%zd", key, objs2.size());
|
|
return false;
|
|
}
|
|
|
|
auto o = objs2[1];
|
|
if (o->get_type() != REDIS_OBJ_STRING) {
|
|
logger_error("invalid object type=%d, key=%s", (int) o->get_type(), key);
|
|
return false;
|
|
}
|
|
|
|
auto v = o->get_str();
|
|
if (v == nullptr || *v == 0) {
|
|
logger_error("value null, key=%s", key);
|
|
return false;
|
|
}
|
|
|
|
result.create_object().set_string(v);
|
|
return true;
|
|
}
|
|
|
|
} // namespace pkv
|