diff options
| author | Evgeniy Khramtsov <ekhramtsov@process-one.net> | 2016-12-11 15:03:37 +0300 |
|---|---|---|
| committer | Evgeniy Khramtsov <ekhramtsov@process-one.net> | 2016-12-11 15:03:37 +0300 |
| commit | 5cc8e807df6994fa6b0e860bbcfe0af8fa7fe19f (patch) | |
| tree | f10816cf358fce8744f87e722667683a623e22ec /src/mod_client_state.erl | |
| parent | Fix reload_config (diff) | |
Initial version of new XMPP stream behaviour (for review)
Diffstat (limited to 'src/mod_client_state.erl')
| -rw-r--r-- | src/mod_client_state.erl | 13 |
1 files changed, 4 insertions, 9 deletions
diff --git a/src/mod_client_state.erl b/src/mod_client_state.erl index 2bae7a4f8..a838088fc 100644 --- a/src/mod_client_state.erl +++ b/src/mod_client_state.erl @@ -260,20 +260,15 @@ queue_take(Stanza, Host, C2SState) -> NewState = set_queue(Rest, C2SState), {NewState, get_stanzas(Selected, Host) ++ [Stanza]}. --spec set_queue(csi_queue(), term()) -> term(). +-spec set_queue(csi_queue(), ejabberd_c2s:state()) -> ejabberd_c2s:state(). set_queue(Queue, C2SState) -> - ejabberd_c2s:set_aux_field(csi_queue, Queue, C2SState). + C2SState#{csi_queue => Queue}. --spec get_queue(term()) -> csi_queue(). +-spec get_queue(ejabberd_c2s:state()) -> csi_queue(). get_queue(C2SState) -> - case ejabberd_c2s:get_aux_field(csi_queue, C2SState) of - {ok, Queue} -> - Queue; - error -> - [] - end. + maps:get(csi_queue, C2SState, []). -spec get_stanzas(csi_queue(), binary()) -> [stanza()]. |