diff options
| author | Christophe Romain <christophe.romain@process-one.net> | 2016-11-22 14:48:01 +0100 |
|---|---|---|
| committer | Christophe Romain <christophe.romain@process-one.net> | 2016-11-22 14:48:01 +0100 |
| commit | fbfbb968727617ae9bbeedc9b3d4adb7636ad5da (patch) | |
| tree | e48dafe80850fe41be32a3ab47f27e6e130b832b /src/mod_announce_riak.erl | |
| parent | Add new xmpp repo as dependency in mix.exs (diff) | |
Improve ODBC import
Diffstat (limited to 'src/mod_announce_riak.erl')
| -rw-r--r-- | src/mod_announce_riak.erl | 12 |
1 files changed, 7 insertions, 5 deletions
diff --git a/src/mod_announce_riak.erl b/src/mod_announce_riak.erl index 879eb196..242adee0 100644 --- a/src/mod_announce_riak.erl +++ b/src/mod_announce_riak.erl @@ -11,7 +11,7 @@ %% API -export([init/2, set_motd_users/2, set_motd/2, delete_motd/1, - get_motd/1, is_motd_user/2, set_motd_user/2, import/2]). + get_motd/1, is_motd_user/2, set_motd_user/2, import/3]). -include("xmpp.hrl"). -include("mod_announce.hrl"). @@ -71,11 +71,13 @@ set_motd_user(LUser, LServer) -> #motd_users{us = {LUser, LServer}}, motd_users_schema(), [{'2i', [{<<"server">>, LServer}]}])}. -import(_LServer, #motd{} = Motd) -> - ejabberd_riak:put(Motd, motd_schema()); -import(_LServer, #motd_users{us = {_, S}} = Users) -> +import(LServer, <<"motd">>, [<<>>, XML, _TimeStamp]) -> + El = fxml_stream:parse_element(XML), + ejabberd_riak:put(#motd{server = LServer, packet = El}, motd_schema()); +import(LServer, <<"motd">>, [LUser, <<>>, _TimeStamp]) -> + Users = #motd_users{us = {LUser, LServer}}, ejabberd_riak:put(Users, motd_users_schema(), - [{'2i', [{<<"server">>, S}]}]). + [{'2i', [{<<"server">>, LServer}]}]). %%%=================================================================== %%% Internal functions |