void
test_func_mongo_sync_auto_reconnect_cache (void)
{
mongo_sync_conn_recovery_cache *cache;
mongo_sync_connection *conn;
bson *b;
mongo_packet *p;
GList *hosts;
gchar *primary_addr;
const gchar *error_msg;
primary_addr = g_strdup_printf ("%s:%d", config.primary_host, config.primary_port);
b = bson_new ();
bson_append_int32 (b, "f_sync_auto_reconnect", 1);
bson_finish (b);
cache = mongo_sync_conn_recovery_cache_new ();
mongo_sync_conn_recovery_cache_seed_add (cache,
config.primary_host,
config.primary_port);
conn = mongo_sync_connect_recovery_cache (cache,
TRUE);
ok (mongo_sync_cmd_insert (conn, config.ns, b, NULL) == TRUE);
shutdown (conn->super.fd, SHUT_RDWR);
sleep (1);
ok (mongo_sync_cmd_insert (conn, config.ns, b, NULL) == FALSE,
"Inserting fails with auto-reconnect turned off, and a broken "
"connection");
error_msg = mongo_sync_conn_get_last_error (conn);
ok (error_msg != NULL, "We have an error msg when insert fails.");
mongo_sync_conn_set_auto_reconnect (conn, TRUE);
ok (mongo_sync_cmd_insert (conn, config.ns, b, NULL) == TRUE,
"Inserting works with auto-reconnect turned on, and a broken "
"connection");
error_msg = mongo_sync_conn_get_last_error (conn);
ok (error_msg == NULL,
"After a succesful insert we shouldn't have an error msg.");
mongo_sync_conn_set_auto_reconnect (conn, FALSE);
shutdown (conn->super.fd, SHUT_RDWR);
sleep (1);
ok (mongo_sync_cmd_insert (conn, config.ns, b, NULL) == FALSE,
"Turning off auto-reconnect works");
skip (!config.secondary_host, 7,
"Secondary host not set up");
shutdown (conn->super.fd, SHUT_RDWR);
sleep (1);
p = mongo_sync_cmd_query (conn, config.ns, 0, 0, 1, b, NULL);
ok (p == NULL,
"Query fails with auto-reconnect turned off");
error_msg = mongo_sync_conn_get_last_error(conn);
ok (error_msg != NULL, "We have an error msg after a failure query.");
mongo_sync_conn_set_auto_reconnect (conn, TRUE);
p = mongo_sync_cmd_query (conn, config.ns, 0, 0, 1, b, NULL);
ok (p != NULL,
"Query does reconnect with auto-reconnect turned on");
ok (mongo_sync_conn_get_last_error(conn) == NULL,
"We shouldn't have any error messages after a successful operation.");
mongo_wire_packet_free (p);
hosts = conn->rs.hosts;
ok (cache->rs.hosts == NULL,
"cache is discarded due to connect replace during auto-reconnect");
ok ((conn->rs.hosts != NULL) &&
(g_list_length (conn->rs.hosts) > 0),
"hosts is filled in mongo_sync_connection");
mongo_sync_disconnect (conn);
ok ((cache->rs.hosts != NULL) &&
(cache->rs.hosts == hosts) &&
(g_list_length (cache->rs.hosts) > 0),
"cache is filled by disconnect()");
mongo_sync_conn_recovery_cache_free (cache);
endskip;
g_free (primary_addr);
}
static worker_insert_result_t
afmongodb_worker_insert (LogThrDestDriver *s, LogMessage *msg)
{
MongoDBDestDriver *self = (MongoDBDestDriver *)s;
gboolean success;
gboolean drop_silently = self->template_options.on_error & ON_ERROR_SILENT;
if (!afmongodb_dd_connect(self, TRUE))
return WORKER_INSERT_RESULT_NOT_CONNECTED;
bson_reset (self->bson);
success = value_pairs_walk(self->vp,
afmongodb_vp_obj_start,
afmongodb_vp_process_value,
afmongodb_vp_obj_end,
msg, self->super.seq_num,
LTZ_SEND,
&self->template_options,
self);
bson_finish (self->bson);
if (!success)
{
if (!drop_silently)
{
msg_error("Failed to format message for MongoDB, dropping message",
evt_tag_value_pairs("message", self->vp, msg,
self->super.seq_num,
LTZ_SEND, &self->template_options),
evt_tag_str("driver", self->super.super.super.id),
NULL);
}
return WORKER_INSERT_RESULT_DROP;
}
else
{
msg_debug("Outgoing message to MongoDB destination",
evt_tag_value_pairs("message", self->vp, msg,
self->super.seq_num,
LTZ_SEND, &self->template_options),
evt_tag_str("driver", self->super.super.super.id),
NULL);
if (!mongo_sync_cmd_insert_n(self->conn, self->ns, 1,
(const bson **)&self->bson))
{
msg_error("Network error while inserting into MongoDB",
evt_tag_int("time_reopen", self->super.time_reopen),
evt_tag_str("reason", mongo_sync_conn_get_last_error(self->conn)),
evt_tag_str("driver", self->super.super.super.id),
NULL);
success = FALSE;
}
}
if (!success && (errno == ENOTCONN))
return WORKER_INSERT_RESULT_NOT_CONNECTED;
if (!success)
return WORKER_INSERT_RESULT_ERROR;
return WORKER_INSERT_RESULT_SUCCESS;
}
