| 1 |
|
-module(mod_pubsub_cache_rdbms). |
| 2 |
|
|
| 3 |
|
-behaviour(mod_pubsub_cache_backend). |
| 4 |
|
|
| 5 |
|
-include("pubsub.hrl"). |
| 6 |
|
-include("jlib.hrl"). |
| 7 |
|
-include("mongoose_logger.hrl"). |
| 8 |
|
|
| 9 |
|
-export([start/1, stop/0]). |
| 10 |
|
|
| 11 |
|
-export([ |
| 12 |
|
upsert_last_item/5, |
| 13 |
|
delete_last_item/2, |
| 14 |
|
get_last_item/2]). |
| 15 |
|
%% ------------------------ Backend start/stop ------------------------ |
| 16 |
|
|
| 17 |
|
-spec start(jid:lserver()) -> ok. |
| 18 |
|
start(Host) -> |
| 19 |
:-( |
InsertFields = [<<"nidx">>, <<"itemid">>, <<"created_luser">>, |
| 20 |
|
<<"created_lserver">>, <<"created_at">>, <<"payload">>], |
| 21 |
:-( |
UpdateFields = [<<"itemid">>, <<"created_luser">>, <<"created_lserver">>, |
| 22 |
|
<<"created_at">>, <<"payload">>], |
| 23 |
:-( |
mongoose_rdbms:prepare(pubsub_get_last_item, pubsub_last_item, [nidx], |
| 24 |
|
<<"SELECT nidx, itemid, created_luser, created_at, created_lserver, payload " |
| 25 |
|
"FROM pubsub_last_item WHERE nidx = ?">>), |
| 26 |
:-( |
mongoose_rdbms:prepare(pubsub_delete_last_item, pubsub_last_item, [nidx], |
| 27 |
|
<<"DELETE FROM pubsub_last_item WHERE nidx = ?">>), |
| 28 |
:-( |
rdbms_queries:prepare_upsert(Host, pubsub_last_item_upsert, pubsub_last_item, |
| 29 |
|
InsertFields, |
| 30 |
|
UpdateFields, |
| 31 |
|
[<<"nidx">>]), |
| 32 |
:-( |
ok. |
| 33 |
|
|
| 34 |
|
-spec stop() -> ok. |
| 35 |
:-( |
stop() -> ok. |
| 36 |
|
|
| 37 |
|
%% ------------------- Pubusub last item ------------------------------ |
| 38 |
|
|
| 39 |
|
-spec upsert_last_item(ServerHost :: jid:lserver(), |
| 40 |
|
Nidx :: mod_pubsub:nodeIdx(), |
| 41 |
|
ItemID :: mod_pubsub:itemId(), |
| 42 |
|
Publisher :: jid:jid(), |
| 43 |
|
Payload :: mod_pubsub:payload()) -> ok | {error, Reason :: term()}. |
| 44 |
|
upsert_last_item(ServerHost, Nidx, ItemID, Publisher, Payload) -> |
| 45 |
:-( |
{ModifiedLUser, ModifiedLServer, |
| 46 |
|
CreatedAt, PayloadBin} = prepare_upsert_params(Publisher, Payload), |
| 47 |
:-( |
UpdateParams = [ItemID, ModifiedLUser, ModifiedLServer, CreatedAt, PayloadBin], |
| 48 |
:-( |
InsertParams = [Nidx | UpdateParams], |
| 49 |
:-( |
UniqueKeyValues = [Nidx], |
| 50 |
:-( |
Res = rdbms_queries:execute_upsert(ServerHost, pubsub_last_item_upsert, |
| 51 |
|
InsertParams, UpdateParams, UniqueKeyValues), |
| 52 |
:-( |
convert_rdbms_response(Res). |
| 53 |
|
|
| 54 |
|
-spec delete_last_item(ServerHost :: binary(), |
| 55 |
|
Nidx :: mod_pubsub:nodeIdx()) -> ok | {error, Reason :: term()}. |
| 56 |
|
delete_last_item(ServerHost, Nidx) -> |
| 57 |
:-( |
Res = mongoose_rdbms:execute_successfully(ServerHost, pubsub_delete_last_item, [Nidx]), |
| 58 |
:-( |
convert_rdbms_response(Res). |
| 59 |
|
|
| 60 |
|
-spec get_last_item(ServerHost :: binary(), |
| 61 |
|
Nidx :: mod_pubsub:nodeIdx()) -> |
| 62 |
|
{ok, LastItem :: mod_pubsub:pubsubLastItem()} | {error, Reason :: term()}. |
| 63 |
|
get_last_item(ServerHost, Nidx) -> |
| 64 |
:-( |
Res = mongoose_rdbms:execute_successfully(ServerHost, pubsub_get_last_item, [Nidx]), |
| 65 |
:-( |
convert_rdbms_response(Res). |
| 66 |
|
|
| 67 |
|
%%==================================================================== |
| 68 |
|
%% Helpers |
| 69 |
|
%%==================================================================== |
| 70 |
|
|
| 71 |
|
convert_rdbms_response({selected, []}) -> |
| 72 |
:-( |
{error, no_items}; |
| 73 |
|
convert_rdbms_response({selected, [SelectedItem]}) -> |
| 74 |
:-( |
LastItem = item_to_record(SelectedItem), |
| 75 |
:-( |
{ok, LastItem}; |
| 76 |
|
convert_rdbms_response({updated, _}) -> |
| 77 |
:-( |
ok; |
| 78 |
|
convert_rdbms_response(Response) -> |
| 79 |
:-( |
?LOG_ERROR(#{what => pubsub_rdbms_cache_failed, reason => Response}), |
| 80 |
:-( |
{error, pubsub_rdbms_cache_failed}. |
| 81 |
|
|
| 82 |
|
prepare_upsert_params(Publisher, Payload) -> |
| 83 |
:-( |
PayloadXML = #xmlel{name = <<"item">>, children = Payload}, |
| 84 |
:-( |
CreatedAt = os:system_time(microsecond), |
| 85 |
:-( |
BinaryPayload = exml:to_binary(PayloadXML), |
| 86 |
:-( |
ModifiedLUser = Publisher#jid.luser, |
| 87 |
:-( |
ModifiedLServer = Publisher#jid.lserver, |
| 88 |
:-( |
{ |
| 89 |
|
ModifiedLUser, ModifiedLServer, |
| 90 |
|
CreatedAt, BinaryPayload |
| 91 |
|
}. |
| 92 |
|
|
| 93 |
|
item_to_record({NodeIdx, ItemId, CreatedLUser, CreatedAt, CreatedLServer, PayloadDB}) -> |
| 94 |
:-( |
PayloadXML = mongoose_rdbms:unescape_binary(global, PayloadDB), |
| 95 |
:-( |
{ok, #xmlel{children = Payload}} = exml:parse(PayloadXML), |
| 96 |
:-( |
Creation = {mongoose_rdbms:result_to_integer(CreatedAt), |
| 97 |
|
{CreatedLUser, CreatedLServer, <<>>}}, |
| 98 |
:-( |
#pubsub_last_item{itemid = ItemId, |
| 99 |
|
nodeid = NodeIdx, |
| 100 |
|
creation = Creation, |
| 101 |
|
payload = Payload}. |