dict-redis.c revision 35f3b7e05afecacd0332c210c6e253911c2813d8
5a580c3a38ced62d4bcc95b8ac7c4f2935b5d294Timo Sirainen/* Copyright (c) 2008-2013 Dovecot authors, see the included COPYING redis */
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen#include "lib.h"
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen#include "array.h"
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen#include "str.h"
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen#include "istream.h"
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen#include "ostream.h"
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen#include "connection.h"
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen#include "dict-private.h"
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen#define REDIS_DEFAULT_PORT 6379
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen#define REDIS_DEFAULT_LOOKUP_TIMEOUT_MSECS (1000*30)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen#define DICT_USERNAME_SEPARATOR '/'
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainenenum redis_input_state {
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen /* expecting $-1 / $<size> followed by GET reply */
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen REDIS_INPUT_STATE_GET,
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen /* expecting +QUEUED */
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen REDIS_INPUT_STATE_MULTI,
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen /* expecting +OK reply for DISCARD */
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen REDIS_INPUT_STATE_DISCARD,
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen /* expecting *<nreplies> */
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen REDIS_INPUT_STATE_EXEC,
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen /* expecting EXEC reply */
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen REDIS_INPUT_STATE_EXEC_REPLY
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen};
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainenstruct redis_connection {
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen struct connection conn;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen struct redis_dict *dict;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen string_t *last_reply;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen unsigned int bytes_left;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen bool value_not_found;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen bool value_received;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen};
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainenstruct redis_dict_reply {
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen unsigned int reply_count;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen dict_transaction_commit_callback_t *callback;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen void *context;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen};
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainenstruct redis_dict {
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen struct dict dict;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen struct ip_addr ip;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen char *username, *key_prefix;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen unsigned int port;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen unsigned int timeout_msecs;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen struct ioloop *ioloop;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen struct redis_connection conn;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
4ee00532a265bdfb38539d811fcd12d51210ac35Timo Sirainen ARRAY(enum redis_input_state) input_states;
4ee00532a265bdfb38539d811fcd12d51210ac35Timo Sirainen ARRAY(struct redis_dict_reply) replies;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
aa8e63009c5aa3866bbf5a3e69a86b1ab480c4ddTimo Sirainen bool connected;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen bool transaction_open;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen};
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainenstruct redis_dict_transaction_context {
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen struct dict_transaction_context ctx;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen unsigned int cmd_count;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen bool failed;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen};
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainenstatic struct connection_list *redis_connections;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainenstatic void
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainenredis_input_state_add(struct redis_dict *dict, enum redis_input_state state)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen{
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen array_append(&dict->input_states, &state, 1);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen}
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainenstatic void redis_input_state_remove(struct redis_dict *dict)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen{
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen array_delete(&dict->input_states, 0, 1);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen}
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainenstatic void redis_conn_destroy(struct connection *_conn)
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen{
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen struct redis_connection *conn = (struct redis_connection *)_conn;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen const struct redis_dict_reply *reply;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen
aa8e63009c5aa3866bbf5a3e69a86b1ab480c4ddTimo Sirainen conn->dict->connected = FALSE;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen connection_disconnect(_conn);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen array_foreach(&conn->dict->replies, reply) {
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen if (reply->callback != NULL)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen reply->callback(-1, reply->context);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen }
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen array_clear(&conn->dict->replies);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen array_clear(&conn->dict->input_states);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen if (conn->dict->ioloop != NULL)
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen io_loop_stop(conn->dict->ioloop);
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen}
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainenstatic void redis_wait(struct redis_dict *dict)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen{
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen struct ioloop *prev_ioloop = current_ioloop;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen i_assert(dict->ioloop == NULL);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen dict->ioloop = io_loop_create();
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen connection_switch_ioloop(&dict->conn.conn);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen do {
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen io_loop_run(dict->ioloop);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen } while (array_count(&dict->input_states) > 0);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
35f3b7e05afecacd0332c210c6e253911c2813d8Timo Sirainen io_loop_set_current(prev_ioloop);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen connection_switch_ioloop(&dict->conn.conn);
35f3b7e05afecacd0332c210c6e253911c2813d8Timo Sirainen io_loop_set_current(dict->ioloop);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen io_loop_destroy(&dict->ioloop);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen}
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
531ddc880d8c8d1d3b4b6c5cf8abb1abb4a2e0daTimo Sirainenstatic int redis_input_get(struct redis_connection *conn)
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen{
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen const unsigned char *data;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen size_t size;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen const char *line;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen if (conn->bytes_left == 0) {
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen /* read the size first */
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen line = i_stream_next_line(conn->conn.input);
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen if (line == NULL)
531ddc880d8c8d1d3b4b6c5cf8abb1abb4a2e0daTimo Sirainen return 0;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen if (strcmp(line, "$-1") == 0) {
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen conn->value_received = TRUE;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen conn->value_not_found = TRUE;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen if (conn->dict->ioloop != NULL)
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen io_loop_stop(conn->dict->ioloop);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen redis_input_state_remove(conn->dict);
531ddc880d8c8d1d3b4b6c5cf8abb1abb4a2e0daTimo Sirainen return 1;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen }
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen if (line[0] != '$' || str_to_uint(line+1, &conn->bytes_left) < 0) {
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen i_error("redis: Unexpected input (wanted $size): %s",
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen line);
c3f05842d124c798afdcf67ccf29d8fe29ef2316Timo Sirainen return -1;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen }
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen conn->bytes_left += 2; /* include trailing CRLF */
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen }
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen data = i_stream_get_data(conn->conn.input, &size);
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen if (size > conn->bytes_left)
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen size = conn->bytes_left;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen str_append_n(conn->last_reply, data, size);
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen conn->bytes_left -= size;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen i_stream_skip(conn->conn.input, size);
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen
531ddc880d8c8d1d3b4b6c5cf8abb1abb4a2e0daTimo Sirainen if (conn->bytes_left > 0)
531ddc880d8c8d1d3b4b6c5cf8abb1abb4a2e0daTimo Sirainen return 0;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
531ddc880d8c8d1d3b4b6c5cf8abb1abb4a2e0daTimo Sirainen /* reply fully read - drop trailing CRLF */
531ddc880d8c8d1d3b4b6c5cf8abb1abb4a2e0daTimo Sirainen conn->value_received = TRUE;
531ddc880d8c8d1d3b4b6c5cf8abb1abb4a2e0daTimo Sirainen str_truncate(conn->last_reply, str_len(conn->last_reply)-2);
531ddc880d8c8d1d3b4b6c5cf8abb1abb4a2e0daTimo Sirainen
531ddc880d8c8d1d3b4b6c5cf8abb1abb4a2e0daTimo Sirainen if (conn->dict->ioloop != NULL)
531ddc880d8c8d1d3b4b6c5cf8abb1abb4a2e0daTimo Sirainen io_loop_stop(conn->dict->ioloop);
531ddc880d8c8d1d3b4b6c5cf8abb1abb4a2e0daTimo Sirainen redis_input_state_remove(conn->dict);
531ddc880d8c8d1d3b4b6c5cf8abb1abb4a2e0daTimo Sirainen return 1;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen}
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainenstatic int redis_conn_input_more(struct redis_connection *conn)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen{
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen struct redis_dict *dict = conn->dict;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen struct redis_dict_reply *reply;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen const enum redis_input_state *states;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen enum redis_input_state state;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen unsigned int count, num_replies;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen const char *line;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen states = array_get(&dict->input_states, &count);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen if (count == 0) {
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen line = i_stream_next_line(conn->conn.input);
531ddc880d8c8d1d3b4b6c5cf8abb1abb4a2e0daTimo Sirainen if (line == NULL)
531ddc880d8c8d1d3b4b6c5cf8abb1abb4a2e0daTimo Sirainen return 0;
531ddc880d8c8d1d3b4b6c5cf8abb1abb4a2e0daTimo Sirainen i_error("redis: Unexpected input (expected nothing): %s", line);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen return -1;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen }
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen state = states[0];
531ddc880d8c8d1d3b4b6c5cf8abb1abb4a2e0daTimo Sirainen if (state == REDIS_INPUT_STATE_GET)
531ddc880d8c8d1d3b4b6c5cf8abb1abb4a2e0daTimo Sirainen return redis_input_get(conn);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen line = i_stream_next_line(conn->conn.input);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen if (line == NULL)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen return 0;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen redis_input_state_remove(dict);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen switch (state) {
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen case REDIS_INPUT_STATE_GET:
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen i_unreached();
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen case REDIS_INPUT_STATE_MULTI:
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen case REDIS_INPUT_STATE_DISCARD:
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen if (line[0] != '+')
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen break;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen return 1;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen case REDIS_INPUT_STATE_EXEC:
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen if (line[0] != '*' || str_to_uint(line+1, &num_replies) < 0)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen break;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen reply = array_idx_modifiable(&dict->replies, 0);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen i_assert(reply->reply_count > 0);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen if (reply->reply_count != num_replies) {
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen i_error("redis: EXEC expected %u replies, not %u",
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen reply->reply_count, num_replies);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen return -1;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen }
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen return 1;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen case REDIS_INPUT_STATE_EXEC_REPLY:
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen if (*line != '+' && *line != ':')
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen break;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen /* success, just ignore the actual reply */
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen reply = array_idx_modifiable(&dict->replies, 0);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen i_assert(reply->reply_count > 0);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen if (--reply->reply_count == 0) {
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen if (reply->callback != NULL)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen reply->callback(1, reply->context);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen array_delete(&dict->replies, 0, 1);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen /* if we're running in a dict-ioloop, we're handling a
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen synchronous commit and need to stop now */
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen if (array_count(&dict->replies) == 0 &&
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen conn->dict->ioloop != NULL)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen io_loop_stop(conn->dict->ioloop);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen }
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen return 1;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen }
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen i_error("redis: Unexpected input (state=%d): %s", state, line);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen return -1;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen}
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainenstatic void redis_conn_input(struct connection *_conn)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen{
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen struct redis_connection *conn = (struct redis_connection *)_conn;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen int ret;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen switch (i_stream_read(_conn->input)) {
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen case 0:
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen return;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen case -1:
9e04a4a3c7105fdaa22ce8af96b4c3529c2a22f5Timo Sirainen if (conn->dict->ioloop != NULL)
9e04a4a3c7105fdaa22ce8af96b4c3529c2a22f5Timo Sirainen i_error("redis: Disconnected unexpectedly");
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen redis_conn_destroy(_conn);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen return;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen default:
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen break;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen }
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
531ddc880d8c8d1d3b4b6c5cf8abb1abb4a2e0daTimo Sirainen while ((ret = redis_conn_input_more(conn)) > 0) ;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen if (ret < 0)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen redis_conn_destroy(_conn);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen}
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
7db7fbea5d8a07463b625f93d69166d56018dadfTimo Sirainenstatic void redis_conn_connected(struct connection *_conn, bool success)
aa8e63009c5aa3866bbf5a3e69a86b1ab480c4ddTimo Sirainen{
aa8e63009c5aa3866bbf5a3e69a86b1ab480c4ddTimo Sirainen struct redis_connection *conn = (struct redis_connection *)_conn;
aa8e63009c5aa3866bbf5a3e69a86b1ab480c4ddTimo Sirainen
7db7fbea5d8a07463b625f93d69166d56018dadfTimo Sirainen if (!success) {
aa8e63009c5aa3866bbf5a3e69a86b1ab480c4ddTimo Sirainen i_error("redis: connect(%s, %u) failed: %m",
aa8e63009c5aa3866bbf5a3e69a86b1ab480c4ddTimo Sirainen net_ip2addr(&conn->dict->ip), conn->dict->port);
aa8e63009c5aa3866bbf5a3e69a86b1ab480c4ddTimo Sirainen } else {
aa8e63009c5aa3866bbf5a3e69a86b1ab480c4ddTimo Sirainen conn->dict->connected = TRUE;
aa8e63009c5aa3866bbf5a3e69a86b1ab480c4ddTimo Sirainen }
aa8e63009c5aa3866bbf5a3e69a86b1ab480c4ddTimo Sirainen if (conn->dict->ioloop != NULL)
aa8e63009c5aa3866bbf5a3e69a86b1ab480c4ddTimo Sirainen io_loop_stop(conn->dict->ioloop);
aa8e63009c5aa3866bbf5a3e69a86b1ab480c4ddTimo Sirainen}
aa8e63009c5aa3866bbf5a3e69a86b1ab480c4ddTimo Sirainen
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainenstatic const struct connection_settings redis_conn_set = {
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen .input_max_size = (size_t)-1,
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen .output_max_size = (size_t)-1,
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen .client = TRUE
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen};
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainenstatic const struct connection_vfuncs redis_conn_vfuncs = {
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen .destroy = redis_conn_destroy,
aa8e63009c5aa3866bbf5a3e69a86b1ab480c4ddTimo Sirainen .input = redis_conn_input,
7db7fbea5d8a07463b625f93d69166d56018dadfTimo Sirainen .client_connected = redis_conn_connected
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen};
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainenstatic const char *redis_escape_username(const char *username)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen{
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen const char *p;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen string_t *str = t_str_new(64);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen for (p = username; *p != '\0'; p++) {
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen switch (*p) {
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen case DICT_USERNAME_SEPARATOR:
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen str_append(str, "\\-");
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen break;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen case '\\':
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen str_append(str, "\\\\");
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen break;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen default:
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen str_append_c(str, *p);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen }
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen }
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen return str_c(str);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen}
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
10399559650f552a23949772be79eb6a80198c5aTimo Sirainenstatic int
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainenredis_dict_init(struct dict *driver, const char *uri,
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen enum dict_data_type value_type ATTR_UNUSED,
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen const char *username,
eca38954bcf972618f6b85932a3690acbd2b673aTimo Sirainen const char *base_dir ATTR_UNUSED, struct dict **dict_r,
eca38954bcf972618f6b85932a3690acbd2b673aTimo Sirainen const char **error_r)
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen{
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen struct redis_dict *dict;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen const char *const *args;
8f2eb1ee9ec07661bd50275da99b5f351972a49aTimo Sirainen int ret = 0;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen if (redis_connections == NULL) {
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen redis_connections =
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen connection_list_init(&redis_conn_set,
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen &redis_conn_vfuncs);
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen }
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen dict = i_new(struct redis_dict, 1);
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen if (net_addr2ip("127.0.0.1", &dict->ip) < 0)
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen i_unreached();
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen dict->port = REDIS_DEFAULT_PORT;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen dict->timeout_msecs = REDIS_DEFAULT_LOOKUP_TIMEOUT_MSECS;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen dict->key_prefix = i_strdup("");
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen args = t_strsplit(uri, ":");
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen for (; *args != NULL; args++) {
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen if (strncmp(*args, "host=", 5) == 0) {
8f2eb1ee9ec07661bd50275da99b5f351972a49aTimo Sirainen if (net_addr2ip(*args+5, &dict->ip) < 0) {
eca38954bcf972618f6b85932a3690acbd2b673aTimo Sirainen *error_r = t_strdup_printf("Invalid IP: %s",
eca38954bcf972618f6b85932a3690acbd2b673aTimo Sirainen *args+5);
8f2eb1ee9ec07661bd50275da99b5f351972a49aTimo Sirainen ret = -1;
8f2eb1ee9ec07661bd50275da99b5f351972a49aTimo Sirainen }
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen } else if (strncmp(*args, "port=", 5) == 0) {
8f2eb1ee9ec07661bd50275da99b5f351972a49aTimo Sirainen if (str_to_uint(*args+5, &dict->port) < 0) {
eca38954bcf972618f6b85932a3690acbd2b673aTimo Sirainen *error_r = t_strdup_printf("Invalid port: %s",
eca38954bcf972618f6b85932a3690acbd2b673aTimo Sirainen *args+5);
8f2eb1ee9ec07661bd50275da99b5f351972a49aTimo Sirainen ret = -1;
8f2eb1ee9ec07661bd50275da99b5f351972a49aTimo Sirainen }
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen } else if (strncmp(*args, "prefix=", 7) == 0) {
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen i_free(dict->key_prefix);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen dict->key_prefix = i_strdup(*args + 7);
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen } else if (strncmp(*args, "timeout_msecs=", 14) == 0) {
8f2eb1ee9ec07661bd50275da99b5f351972a49aTimo Sirainen if (str_to_uint(*args+14, &dict->timeout_msecs) < 0) {
eca38954bcf972618f6b85932a3690acbd2b673aTimo Sirainen *error_r = t_strdup_printf(
eca38954bcf972618f6b85932a3690acbd2b673aTimo Sirainen "Invalid timeout_msecs: %s", *args+14);
8f2eb1ee9ec07661bd50275da99b5f351972a49aTimo Sirainen ret = -1;
8f2eb1ee9ec07661bd50275da99b5f351972a49aTimo Sirainen }
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen } else {
eca38954bcf972618f6b85932a3690acbd2b673aTimo Sirainen *error_r = t_strdup_printf("Unknown parameter: %s",
eca38954bcf972618f6b85932a3690acbd2b673aTimo Sirainen *args);
8f2eb1ee9ec07661bd50275da99b5f351972a49aTimo Sirainen ret = -1;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen }
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen }
8f2eb1ee9ec07661bd50275da99b5f351972a49aTimo Sirainen if (ret < 0) {
8f2eb1ee9ec07661bd50275da99b5f351972a49aTimo Sirainen i_free(dict->key_prefix);
8f2eb1ee9ec07661bd50275da99b5f351972a49aTimo Sirainen i_free(dict);
8f2eb1ee9ec07661bd50275da99b5f351972a49aTimo Sirainen return -1;
8f2eb1ee9ec07661bd50275da99b5f351972a49aTimo Sirainen }
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen connection_init_client_ip(redis_connections, &dict->conn.conn,
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen &dict->ip, dict->port);
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen dict->dict = *driver;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen dict->conn.last_reply = str_new(default_pool, 256);
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen dict->conn.dict = dict;
8f2eb1ee9ec07661bd50275da99b5f351972a49aTimo Sirainen
8f2eb1ee9ec07661bd50275da99b5f351972a49aTimo Sirainen i_array_init(&dict->input_states, 4);
8f2eb1ee9ec07661bd50275da99b5f351972a49aTimo Sirainen i_array_init(&dict->replies, 4);
8f2eb1ee9ec07661bd50275da99b5f351972a49aTimo Sirainen if (strchr(username, DICT_USERNAME_SEPARATOR) == NULL)
8f2eb1ee9ec07661bd50275da99b5f351972a49aTimo Sirainen dict->username = i_strdup(username);
8f2eb1ee9ec07661bd50275da99b5f351972a49aTimo Sirainen else {
8f2eb1ee9ec07661bd50275da99b5f351972a49aTimo Sirainen /* escape the username */
8f2eb1ee9ec07661bd50275da99b5f351972a49aTimo Sirainen dict->username = i_strdup(redis_escape_username(username));
8f2eb1ee9ec07661bd50275da99b5f351972a49aTimo Sirainen }
8f2eb1ee9ec07661bd50275da99b5f351972a49aTimo Sirainen
10399559650f552a23949772be79eb6a80198c5aTimo Sirainen *dict_r = &dict->dict;
8f2eb1ee9ec07661bd50275da99b5f351972a49aTimo Sirainen return 0;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen}
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainenstatic void redis_dict_deinit(struct dict *_dict)
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen{
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen struct redis_dict *dict = (struct redis_dict *)_dict;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen if (array_count(&dict->input_states) > 0) {
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen i_assert(dict->connected);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen redis_wait(dict);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen }
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen connection_deinit(&dict->conn.conn);
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen str_free(&dict->conn.last_reply);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen array_free(&dict->replies);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen array_free(&dict->input_states);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen i_free(dict->key_prefix);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen i_free(dict->username);
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen i_free(dict);
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen if (redis_connections->connections == NULL)
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen connection_list_deinit(&redis_connections);
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen}
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainenstatic void redis_dict_lookup_timeout(struct redis_dict *dict)
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen{
aa8e63009c5aa3866bbf5a3e69a86b1ab480c4ddTimo Sirainen i_error("redis: Lookup timed out in %u.%03u secs",
aa8e63009c5aa3866bbf5a3e69a86b1ab480c4ddTimo Sirainen dict->timeout_msecs/1000, dict->timeout_msecs%1000);
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen io_loop_stop(dict->ioloop);
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen}
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainenstatic const char *
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainenredis_dict_get_full_key(struct redis_dict *dict, const char *key)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen{
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen if (strncmp(key, DICT_PATH_SHARED, strlen(DICT_PATH_SHARED)) == 0)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen key += strlen(DICT_PATH_SHARED);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen else if (strncmp(key, DICT_PATH_PRIVATE, strlen(DICT_PATH_PRIVATE)) == 0) {
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen key = t_strdup_printf("%s%c%s", dict->username,
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen DICT_USERNAME_SEPARATOR,
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen key + strlen(DICT_PATH_PRIVATE));
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen } else {
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen i_unreached();
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen }
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen if (*dict->key_prefix != '\0')
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen key = t_strconcat(dict->key_prefix, key, NULL);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen return key;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen}
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainenstatic int
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainenredis_dict_lookup_real(struct redis_dict *dict, pool_t pool,
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen const char *key, const char **value_r)
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen{
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen struct timeout *to;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen const char *cmd;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen struct ioloop *prev_ioloop = current_ioloop;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen key = redis_dict_get_full_key(dict, key);
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen dict->conn.value_received = FALSE;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen dict->conn.value_not_found = FALSE;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen i_assert(dict->ioloop == NULL);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen dict->ioloop = io_loop_create();
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen connection_switch_ioloop(&dict->conn.conn);
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen if (dict->conn.conn.fd_in == -1 &&
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen connection_client_connect(&dict->conn.conn) < 0) {
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen i_error("redis: Couldn't connect to %s:%u",
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen net_ip2addr(&dict->ip), dict->port);
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen } else {
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen to = timeout_add(dict->timeout_msecs,
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen redis_dict_lookup_timeout, dict);
aa8e63009c5aa3866bbf5a3e69a86b1ab480c4ddTimo Sirainen if (!dict->connected) {
aa8e63009c5aa3866bbf5a3e69a86b1ab480c4ddTimo Sirainen /* wait for connection */
aa8e63009c5aa3866bbf5a3e69a86b1ab480c4ddTimo Sirainen io_loop_run(dict->ioloop);
aa8e63009c5aa3866bbf5a3e69a86b1ab480c4ddTimo Sirainen }
aa8e63009c5aa3866bbf5a3e69a86b1ab480c4ddTimo Sirainen
aa8e63009c5aa3866bbf5a3e69a86b1ab480c4ddTimo Sirainen if (dict->connected) {
aa8e63009c5aa3866bbf5a3e69a86b1ab480c4ddTimo Sirainen cmd = t_strdup_printf("*2\r\n$3\r\nGET\r\n$%d\r\n%s\r\n",
aa8e63009c5aa3866bbf5a3e69a86b1ab480c4ddTimo Sirainen (int)strlen(key), key);
1d2c463d23f09f15727edae9c78b07ec6a7a27daTimo Sirainen o_stream_nsend_str(dict->conn.conn.output, cmd);
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen
aa8e63009c5aa3866bbf5a3e69a86b1ab480c4ddTimo Sirainen str_truncate(dict->conn.last_reply, 0);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen redis_input_state_add(dict, REDIS_INPUT_STATE_GET);
97d5d23d41aab07e0e7d98c1f9b834e325d7d1b3Timo Sirainen do {
97d5d23d41aab07e0e7d98c1f9b834e325d7d1b3Timo Sirainen io_loop_run(dict->ioloop);
97d5d23d41aab07e0e7d98c1f9b834e325d7d1b3Timo Sirainen } while (array_count(&dict->input_states) > 0);
aa8e63009c5aa3866bbf5a3e69a86b1ab480c4ddTimo Sirainen }
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen timeout_remove(&to);
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen }
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen
35f3b7e05afecacd0332c210c6e253911c2813d8Timo Sirainen io_loop_set_current(prev_ioloop);
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen connection_switch_ioloop(&dict->conn.conn);
35f3b7e05afecacd0332c210c6e253911c2813d8Timo Sirainen io_loop_set_current(dict->ioloop);
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen io_loop_destroy(&dict->ioloop);
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen if (!dict->conn.value_received) {
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen /* we failed in some way. make sure we disconnect since the
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen connection state isn't known anymore */
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen redis_conn_destroy(&dict->conn.conn);
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen return -1;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen }
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen if (dict->conn.value_not_found)
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen return 0;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen *value_r = p_strdup(pool, str_c(dict->conn.last_reply));
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen return 1;
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen}
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainenstatic int redis_dict_lookup(struct dict *_dict, pool_t pool,
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen const char *key, const char **value_r)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen{
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen struct redis_dict *dict = (struct redis_dict *)_dict;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen int ret;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen i_assert(!dict->transaction_open);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen if (pool->datastack_pool)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen ret = redis_dict_lookup_real(dict, pool, key, value_r);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen else T_BEGIN {
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen ret = redis_dict_lookup_real(dict, pool, key, value_r);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen } T_END;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen return ret;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen}
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainenstatic struct dict_transaction_context *
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainenredis_transaction_init(struct dict *_dict)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen{
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen struct redis_dict *dict = (struct redis_dict *)_dict;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen struct redis_dict_transaction_context *ctx;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen i_assert(!dict->transaction_open);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen dict->transaction_open = TRUE;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen ctx = i_new(struct redis_dict_transaction_context, 1);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen ctx->ctx.dict = _dict;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen if (dict->conn.conn.fd_in == -1 &&
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen connection_client_connect(&dict->conn.conn) < 0) {
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen i_error("redis: Couldn't connect to %s:%u",
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen net_ip2addr(&dict->ip), dict->port);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen } else if (!dict->connected) {
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen /* wait for connection */
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen redis_wait(dict);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen }
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen return &ctx->ctx;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen}
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainenstatic int
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainenredis_transaction_commit(struct dict_transaction_context *_ctx, bool async,
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen dict_transaction_commit_callback_t *callback,
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen void *context)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen{
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen struct redis_dict_transaction_context *ctx =
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen (struct redis_dict_transaction_context *)_ctx;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen struct redis_dict *dict = (struct redis_dict *)_ctx->dict;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen struct redis_dict_reply *reply;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen unsigned int i;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen int ret = 1;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen i_assert(dict->transaction_open);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen dict->transaction_open = FALSE;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen if (ctx->failed) {
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen /* make sure we're disconnected */
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen redis_conn_destroy(&dict->conn.conn);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen ret = -1;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen } else if (_ctx->changed) {
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen i_assert(ctx->cmd_count > 0);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1d2c463d23f09f15727edae9c78b07ec6a7a27daTimo Sirainen o_stream_nsend_str(dict->conn.conn.output,
1d2c463d23f09f15727edae9c78b07ec6a7a27daTimo Sirainen "*1\r\n$4\r\nEXEC\r\n");
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen reply = array_append_space(&dict->replies);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen reply->callback = callback;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen reply->context = context;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen reply->reply_count = ctx->cmd_count;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen redis_input_state_add(dict, REDIS_INPUT_STATE_EXEC);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen for (i = 0; i < ctx->cmd_count; i++)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen redis_input_state_add(dict, REDIS_INPUT_STATE_EXEC_REPLY);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen if (async)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen return 1;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen redis_wait(dict);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen }
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen if (callback != NULL)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen callback(ret, context);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen i_free(ctx);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen return ret;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen}
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainenstatic void redis_transaction_rollback(struct dict_transaction_context *_ctx)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen{
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen struct redis_dict_transaction_context *ctx =
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen (struct redis_dict_transaction_context *)_ctx;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen struct redis_dict *dict = (struct redis_dict *)_ctx->dict;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen struct redis_dict_reply *reply;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen i_assert(dict->transaction_open);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen dict->transaction_open = FALSE;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen if (ctx->failed) {
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen /* make sure we're disconnected */
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen redis_conn_destroy(&dict->conn.conn);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen } else if (_ctx->changed) {
1d2c463d23f09f15727edae9c78b07ec6a7a27daTimo Sirainen o_stream_nsend_str(dict->conn.conn.output,
1d2c463d23f09f15727edae9c78b07ec6a7a27daTimo Sirainen "*1\r\n$7\r\nDISCARD\r\n");
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen reply = array_append_space(&dict->replies);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen reply->reply_count = 1;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen redis_input_state_add(dict, REDIS_INPUT_STATE_DISCARD);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen }
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen i_free(ctx);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen}
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainenstatic int redis_check_transaction(struct redis_dict_transaction_context *ctx)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen{
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen struct redis_dict *dict = (struct redis_dict *)ctx->ctx.dict;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen if (ctx->failed)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen return -1;
6ac32bfcdcd9e96e9b6614914fc4b0a926dcfa69Timo Sirainen if (!dict->connected) {
6ac32bfcdcd9e96e9b6614914fc4b0a926dcfa69Timo Sirainen /* disconnected during transaction */
6ac32bfcdcd9e96e9b6614914fc4b0a926dcfa69Timo Sirainen ctx->failed = TRUE;
6ac32bfcdcd9e96e9b6614914fc4b0a926dcfa69Timo Sirainen return -1;
6ac32bfcdcd9e96e9b6614914fc4b0a926dcfa69Timo Sirainen }
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen if (ctx->ctx.changed)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen return 0;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen redis_input_state_add(dict, REDIS_INPUT_STATE_MULTI);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen if (o_stream_send_str(dict->conn.conn.output,
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen "*1\r\n$5\r\nMULTI\r\n") < 0) {
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen ctx->failed = TRUE;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen return -1;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen }
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen return 0;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen}
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainenstatic void redis_set(struct dict_transaction_context *_ctx,
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen const char *key, const char *value)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen{
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen struct redis_dict_transaction_context *ctx =
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen (struct redis_dict_transaction_context *)_ctx;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen struct redis_dict *dict = (struct redis_dict *)_ctx->dict;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen const char *cmd;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen if (redis_check_transaction(ctx) < 0)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen return;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen key = redis_dict_get_full_key(dict, key);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen cmd = t_strdup_printf("*3\r\n$3\r\nSET\r\n$%u\r\n%s\r\n$%u\r\n%s\r\n",
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen (unsigned int)strlen(key), key,
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen (unsigned int)strlen(value), value);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen if (o_stream_send_str(dict->conn.conn.output, cmd) < 0)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen ctx->failed = TRUE;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen redis_input_state_add(dict, REDIS_INPUT_STATE_MULTI);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen ctx->cmd_count++;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen}
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainenstatic void redis_unset(struct dict_transaction_context *_ctx,
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen const char *key)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen{
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen struct redis_dict_transaction_context *ctx =
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen (struct redis_dict_transaction_context *)_ctx;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen struct redis_dict *dict = (struct redis_dict *)_ctx->dict;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen const char *cmd;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen if (redis_check_transaction(ctx) < 0)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen return;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen key = redis_dict_get_full_key(dict, key);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen cmd = t_strdup_printf("*2\r\n$3\r\nDEL\r\n$%u\r\n%s\r\n",
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen (unsigned int)strlen(key), key);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen if (o_stream_send_str(dict->conn.conn.output, cmd) < 0)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen ctx->failed = TRUE;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen redis_input_state_add(dict, REDIS_INPUT_STATE_MULTI);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen ctx->cmd_count++;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen}
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
2aed6c2062317d1750f59c5c88e77d9f10967462Timo Sirainenstatic void redis_append(struct dict_transaction_context *_ctx,
2aed6c2062317d1750f59c5c88e77d9f10967462Timo Sirainen const char *key, const char *value)
2aed6c2062317d1750f59c5c88e77d9f10967462Timo Sirainen{
2aed6c2062317d1750f59c5c88e77d9f10967462Timo Sirainen struct redis_dict_transaction_context *ctx =
2aed6c2062317d1750f59c5c88e77d9f10967462Timo Sirainen (struct redis_dict_transaction_context *)_ctx;
2aed6c2062317d1750f59c5c88e77d9f10967462Timo Sirainen struct redis_dict *dict = (struct redis_dict *)_ctx->dict;
2aed6c2062317d1750f59c5c88e77d9f10967462Timo Sirainen const char *cmd;
2aed6c2062317d1750f59c5c88e77d9f10967462Timo Sirainen
2aed6c2062317d1750f59c5c88e77d9f10967462Timo Sirainen if (redis_check_transaction(ctx) < 0)
2aed6c2062317d1750f59c5c88e77d9f10967462Timo Sirainen return;
2aed6c2062317d1750f59c5c88e77d9f10967462Timo Sirainen
2aed6c2062317d1750f59c5c88e77d9f10967462Timo Sirainen key = redis_dict_get_full_key(dict, key);
2aed6c2062317d1750f59c5c88e77d9f10967462Timo Sirainen cmd = t_strdup_printf("*3\r\n$6\r\nAPPEND\r\n$%u\r\n%s\r\n$%u\r\n%s\r\n",
2aed6c2062317d1750f59c5c88e77d9f10967462Timo Sirainen (unsigned int)strlen(key), key,
2aed6c2062317d1750f59c5c88e77d9f10967462Timo Sirainen (unsigned int)strlen(value), value);
2aed6c2062317d1750f59c5c88e77d9f10967462Timo Sirainen if (o_stream_send_str(dict->conn.conn.output, cmd) < 0)
2aed6c2062317d1750f59c5c88e77d9f10967462Timo Sirainen ctx->failed = TRUE;
2aed6c2062317d1750f59c5c88e77d9f10967462Timo Sirainen redis_input_state_add(dict, REDIS_INPUT_STATE_MULTI);
2aed6c2062317d1750f59c5c88e77d9f10967462Timo Sirainen ctx->cmd_count++;
2aed6c2062317d1750f59c5c88e77d9f10967462Timo Sirainen}
2aed6c2062317d1750f59c5c88e77d9f10967462Timo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainenstatic void redis_atomic_inc(struct dict_transaction_context *_ctx,
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen const char *key, long long diff)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen{
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen struct redis_dict_transaction_context *ctx =
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen (struct redis_dict_transaction_context *)_ctx;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen struct redis_dict *dict = (struct redis_dict *)_ctx->dict;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen const char *cmd, *diffstr;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen if (redis_check_transaction(ctx) < 0)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen return;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen key = redis_dict_get_full_key(dict, key);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen diffstr = t_strdup_printf("%lld", diff);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen cmd = t_strdup_printf("*3\r\n$6\r\nINCRBY\r\n$%u\r\n%s\r\n$%u\r\n%s\r\n",
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen (unsigned int)strlen(key), key,
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen (unsigned int)strlen(diffstr), diffstr);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen if (o_stream_send_str(dict->conn.conn.output, cmd) < 0)
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen ctx->failed = TRUE;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen redis_input_state_add(dict, REDIS_INPUT_STATE_MULTI);
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen ctx->cmd_count++;
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen}
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainenstruct dict dict_driver_redis = {
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen .name = "redis",
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen {
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen redis_dict_init,
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen redis_dict_deinit,
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen NULL,
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen redis_dict_lookup,
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen NULL,
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen NULL,
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen NULL,
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen redis_transaction_init,
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen redis_transaction_commit,
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen redis_transaction_rollback,
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen redis_set,
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen redis_unset,
2aed6c2062317d1750f59c5c88e77d9f10967462Timo Sirainen redis_append,
1c7df2f6d12f5e7f0cf1e73452d4ae216a3d092cTimo Sirainen redis_atomic_inc
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen }
31257b47d47510ceb093a6b218810a1a5b830c55Timo Sirainen};