void TypeLoader::OnResult(const Reply &r)
{
if (r.type != Reply::MULTI_BULK || !me->redis)
{
delete this;
return;
}
for (unsigned i = 0; i < r.multi_bulk.size(); ++i)
{
const Reply *reply = r.multi_bulk[i];
if (reply->type != Reply::BULK)
continue;
int64_t id;
try
{
id = convertTo<int64_t>(reply->bulk);
}
catch (const ConvertException &)
{
continue;
}
Serialize::Object *obj = type->Require(id);
if (obj == nullptr)
{
Log(LOG_DEBUG) << "redis: Unable to require object #" << id << " of type " << type->GetName();
continue;
}
std::vector<Anope::string> args = { "SMEMBERS", "keys:" + stringify(id) };
me->redis->SendCommand(new ObjectLoader(me, obj), args);
}
delete this;
}
EventReturn OnLoadDatabase() override
{
const Anope::string &db_name = Anope::DataDir + "/" + Config->GetModule(this)->Get<Anope::string>("database", "anope.db");
std::fstream fd(db_name.c_str(), std::ios_base::in | std::ios_base::binary);
if (!fd.is_open())
{
Log(this) << "Unable to open " << db_name << " for reading!";
return EVENT_STOP;
}
Serialize::TypeBase *type = nullptr;
Serialize::Object *obj = nullptr;
for (Anope::string buf; std::getline(fd, buf.str());)
{
if (buf.find("OBJECT ") == 0)
{
Anope::string t = buf.substr(7);
if (obj)
Log(LOG_DEBUG) << "obj != null but got OBJECT";
if (type)
Log(LOG_DEBUG) << "type != null but got OBJECT";
type = Serialize::TypeBase::Find(t);
obj = nullptr;
}
else if (buf.find("ID ") == 0)
{
if (!type || obj)
continue;
try
{
Serialize::ID id = convertTo<Serialize::ID>(buf.substr(3));
obj = type->Require(id);
}
catch (const ConvertException &)
{
Log(LOG_DEBUG) << "Unable to parse object id " << buf.substr(3);
}
}
else if (buf.find("DATA ") == 0)
{
if (!type)
continue;
if (!obj)
obj = type->Create();
size_t sp = buf.find(' ', 5); // Skip DATA
if (sp == Anope::string::npos)
continue;
Anope::string key = buf.substr(5, sp - 5), value = buf.substr(sp + 1);
Serialize::FieldBase *field = type->GetField(key);
if (field)
field->UnserializeFromString(obj, value);
}
else if (buf.find("END") == 0)
{
type = nullptr;
obj = nullptr;
}
}
fd.close();
loaded = true;
return EVENT_STOP;
}
void SubscriptionListener::OnResult(const Reply &r)
{
/*
* message
* anope
* message
*
* set 4 email [email protected]
* unset 4 email
* create 4 NickCore
* delete 4
*/
const Anope::string &message = r.multi_bulk[2]->bulk;
Anope::string command;
spacesepstream sep(message);
sep.GetToken(command);
if (command == "set" || command == "unset")
{
Anope::string sid, key, value;
sep.GetToken(sid);
sep.GetToken(key);
value = sep.GetRemaining();
Serialize::ID id;
try
{
id = convertTo<Serialize::ID>(sid);
}
catch (const ConvertException &ex)
{
Log(LOG_DEBUG) << "redis: unable to get id for SL update key " << sid;
return;
}
Serialize::Object *obj = Serialize::GetID(id);
if (obj == nullptr)
{
Log(LOG_DEBUG) << "redis: pmessage for unknown object #" << id;
return;
}
Serialize::FieldBase *field = obj->GetSerializableType()->GetField(key);
if (field == nullptr)
{
Log(LOG_DEBUG) << "redis: pmessage for unknown field of object #" << id << ": " << key;
return;
}
Log(LOG_DEBUG_2) << "redis: Setting field " << field->serialize_name << " of object #" << obj->id << " of type " << obj->GetSerializableType()->GetName() << " to " << value;
field->UnserializeFromString(obj, value);
}
else if (command == "create")
{
Anope::string sid, stype;
sep.GetToken(sid);
sep.GetToken(stype);
Serialize::ID id;
try
{
id = convertTo<Serialize::ID>(sid);
}
catch (const ConvertException &ex)
{
Log(LOG_DEBUG) << "redis: unable to get id for SL update key " << sid;
return;
}
Serialize::TypeBase *type = Serialize::TypeBase::Find(stype);
if (type == nullptr)
{
Log(LOG_DEBUG) << "redis: pmessage create for nonexistant type " << stype;
return;
}
Serialize::Object *obj = type->Require(id);
if (obj == nullptr)
{
Log(LOG_DEBUG) << "redis: require for pmessage create type " << type->GetName() << " id #" << id << " returned nullptr";
return;
}
}
else if (command == "delete")
{
Anope::string sid;
sep.GetToken(sid);
Serialize::ID id;
try
{
id = convertTo<Serialize::ID>(sid);
}
catch (const ConvertException &ex)
{
Log(LOG_DEBUG) << "redis: unable to get id for SL update key " << sid;
return;
}
Serialize::Object *obj = Serialize::GetID(id);
if (obj == nullptr)
{
Log(LOG_DEBUG) << "redis: message for unknown object #" << id;
return;
}
obj->Delete();
}
else
Log(LOG_DEBUG) << "redis: unknown message: " << message;
}