static void mongo_print_connection_info(mongo_con_manager *manager, mongo_connection *con, int level)
{
int i;
mongo_manager_log(manager, MLOG_RS, level,
"- connection: type: %s, socket: %d, ping: %d, hash: %s",
mongo_connection_type(con->connection_type),
42, /* FIXME: STREAMS: Maybe we do need a union here: con->socket, */
con->ping_ms,
con->hash
);
for (i = 0; i < con->tag_count; i++) {
mongo_manager_log(manager, MLOG_RS, level,
" - tag: %s", con->tags[i]
);
}
}
static void mongo_print_connection_info(mongo_con_manager *manager, mongo_connection *con, int level)
{
int i;
mongo_manager_log(manager, MLOG_RS, level,
"- connection: type: %s, socket: %d, ping: %d, hash: %s",
mongo_connection_type(con->connection_type),
con->socket,
con->ping_ms,
con->hash
);
for (i = 0; i < con->tag_count; i++) {
mongo_manager_log(manager, MLOG_RS, level,
" - tag: %s", con->tags[i]
);
}
}
/* Sends an ismaster command to the server to find server flags
*
* Returns 1 when it worked, and 0 when an error was encountered. */
int mongo_connection_get_server_flags(mongo_con_manager *manager, mongo_connection *con, mongo_server_options *options, char **error_message)
{
mcon_str *packet;
int32_t max_bson_size = 0, max_message_size = 0;
char *data_buffer;
char *ptr;
char *tags;
char *msg; /* If set and its value is "isdbgrid", it signals we connected to a mongos */
char *set = NULL; /* check if it is a replicaset */
unsigned char ismaster = 0, secondary = 0, arbiter = 0;
mongo_manager_log(manager, MLOG_CON, MLOG_INFO, "get_server_flags: start");
packet = bson_create_ismaster_packet(con);
if (!mongo_connect_send_packet(manager, con, options, packet, &data_buffer, error_message)) {
return 0;
}
/* Find data fields */
ptr = data_buffer + sizeof(int32_t); /* Skip the length */
/* Find max bson size */
if (bson_find_field_as_int32(ptr, "maxBsonObjectSize", &max_bson_size)) {
mongo_manager_log(manager, MLOG_CON, MLOG_FINE, "get_server_flags: setting maxBsonObjectSize to %d", max_bson_size);
con->max_bson_size = max_bson_size;
} else {
/* This seems to be a pre-1.8 MongoDB installation, where we need to
* default to 4MB */
con->max_bson_size = 4194304;
mongo_manager_log(manager, MLOG_CON, MLOG_FINE, "get_server_flags: can't find maxBsonObjectSize, defaulting to %d", con->max_bson_size);
}
/* Find max message size */
if (bson_find_field_as_int32(ptr, "maxMessageSizeBytes", &max_message_size)) {
mongo_manager_log(manager, MLOG_CON, MLOG_FINE, "get_server_flags: setting maxMessageSizeBytes to %d", max_message_size);
con->max_message_size = max_message_size;
} else {
/* This seems to be a pre-2.4 MongoDB installation, where we need to
* default to two times the max BSON size */
con->max_message_size = 2 * con->max_bson_size;
mongo_manager_log(manager, MLOG_CON, MLOG_FINE, "get_server_flags: can't find maxMessageSizeBytes, defaulting to %d", con->max_message_size);
}
/* Check for flags */
bson_find_field_as_bool(ptr, "ismaster", &ismaster);
bson_find_field_as_bool(ptr, "secondary", &secondary);
bson_find_field_as_bool(ptr, "arbiterOnly", &arbiter);
/* Set connection type depending on flags */
if (ismaster) {
/* Find msg and whether it contains "isdbgrid" */
if (bson_find_field_as_string(ptr, "msg", (char**) &msg) && strcmp(msg, "isdbgrid") == 0) {
mongo_manager_log(manager, MLOG_CON, MLOG_FINE, "get_server_flags: msg contains 'isdbgrid' - we're connected to a mongos");
con->connection_type = MONGO_NODE_MONGOS;
} else if (bson_find_field_as_string(ptr, "setName", &set)) {
mongo_manager_log(manager, MLOG_CON, MLOG_FINE, "get_server_flags: got replicaset name: %s", set);
con->connection_type = MONGO_NODE_PRIMARY;
} else {
con->connection_type = MONGO_NODE_STANDALONE;
}
} else if (secondary) {
con->connection_type = MONGO_NODE_SECONDARY;
} else if (arbiter) {
con->connection_type = MONGO_NODE_ARBITER;
} else {
con->connection_type = MONGO_NODE_INVALID;
}
mongo_manager_log(manager, MLOG_CON, MLOG_INFO, "get_server_flags: found server type: %s", mongo_connection_type(con->connection_type));
/* Find read preferences tags */
con->tag_count = 0;
con->tags = NULL;
if (bson_find_field_as_document(ptr, "tags", (char**) &tags)) {
char *it, *name, *value;
int length;
it = tags;
while (bson_array_find_next_string(&it, &name, &value)) {
con->tags = (char**) realloc(con->tags, (con->tag_count + 1) * sizeof(char*));
length = strlen(name) + strlen(value) + 2;
con->tags[con->tag_count] = (char*) malloc(length);
snprintf(con->tags[con->tag_count], length, "%s:%s", name, value);
free(name);
mongo_manager_log(manager, MLOG_CON, MLOG_FINE, "get_server_flags: added tag %s", con->tags[con->tag_count]);
con->tag_count++;
}
}
free(data_buffer);
return 1;
}
