void mongo_servers_copy(mongo_servers *to, mongo_servers *from, int flags)
{
int i;
to->count = from->count;
for (i = 0; i < from->count; i++) {
to->server[i] = calloc(1, sizeof(mongo_server_def));
mongo_server_def_copy(to->server[i], from->server[i], flags);
}
to->options.con_type = from->options.con_type;
if (from->options.repl_set_name) {
to->options.repl_set_name = strdup(from->options.repl_set_name);
}
to->options.connectTimeoutMS = from->options.connectTimeoutMS;
to->options.default_w = from->options.default_w;
to->options.default_wtimeout = from->options.default_wtimeout;
if (from->options.default_wstring) {
to->options.default_wstring = strdup(from->options.default_wstring);
}
mongo_read_preference_copy(&from->read_pref, &to->read_pref);
}
void php_mongo_db_construct(zval *z_client, zval *zlink, char *name, int name_len TSRMLS_DC)
{
mongo_db *db;
mongoclient *link;
if (!php_mongo_db_is_valid_dbname(name, name_len TSRMLS_CC)) {
return;
}
db = (mongo_db*)zend_object_store_get_object(z_client TSRMLS_CC);
db->link = zlink;
zval_add_ref(&db->link);
link = (mongoclient*)zend_object_store_get_object(zlink TSRMLS_CC);
if (!(link->servers)) {
zend_throw_exception(mongo_ce_Exception, "The MongoDB object has not been correctly initialized by its constructor", 0 TSRMLS_CC);
return;
}
if (link->servers->options.default_w != -1) {
zend_update_property_long(mongo_ce_DB, z_client, "w", strlen("w"), link->servers->options.default_w TSRMLS_CC);
} else if (link->servers->options.default_wstring != NULL) {
zend_update_property_string(mongo_ce_DB, z_client, "w", strlen("w"), link->servers->options.default_wstring TSRMLS_CC);
}
if (link->servers->options.default_wtimeout != -1) {
zend_update_property_long(mongo_ce_DB, z_client, "wtimeout", strlen("wtimeout"), link->servers->options.default_wtimeout TSRMLS_CC);
}
mongo_read_preference_copy(&link->servers->read_pref, &db->read_pref);
MAKE_STD_ZVAL(db->name);
ZVAL_STRING(db->name, name, 1);
}
static mongo_connection *mongo_get_read_write_connection_replicaset(mongo_con_manager *manager, mongo_servers *servers, int connection_flags, char **error_message)
{
mongo_connection *con = NULL;
mongo_connection *tmp;
mcon_collection *collection;
char *con_error_message = NULL;
int i;
int found_connected_server = 0;
/* Create a connection to all of the servers in the seed list */
for (i = 0; i < servers->count; i++) {
tmp = mongo_get_connection_single(manager, servers->server[i], &servers->options, connection_flags, (char **) &con_error_message);
if (tmp) {
found_connected_server = 1;
} else if (!(connection_flags & MONGO_CON_FLAG_DONT_CONNECT)) {
mongo_manager_log(manager, MLOG_CON, MLOG_WARN, "Couldn't connect to '%s:%d': %s", servers->server[i]->host, servers->server[i]->port, con_error_message);
free(con_error_message);
}
}
if (!found_connected_server && (connection_flags & MONGO_CON_FLAG_DONT_CONNECT)) {
return NULL;
}
/* Discover more nodes. This also adds a connection to "servers" for each
* new node */
mongo_discover_topology(manager, servers);
/* Depending on whether we want a read or a write connection, run the correct algorithms */
if (connection_flags & MONGO_CON_FLAG_WRITE) {
mongo_read_preference tmp_rp;
mongo_read_preference_copy(&servers->read_pref, &tmp_rp);
tmp_rp.type = MONGO_RP_PRIMARY;
collection = mongo_find_candidate_servers(manager, &tmp_rp, servers);
mongo_read_preference_dtor(&tmp_rp);
} else {
collection = mongo_find_candidate_servers(manager, &servers->read_pref, servers);
}
if (!collection) {
*error_message = strdup("No candidate servers found");
return NULL;
}
if (collection->count == 0) {
*error_message = strdup("No candidate servers found");
mcon_collection_free(collection);
return NULL;
}
collection = mongo_sort_servers(manager, collection, &servers->read_pref);
collection = mongo_select_nearest_servers(manager, collection, &servers->read_pref);
con = mongo_pick_server_from_set(manager, collection, &servers->read_pref);
/* Cleaning up */
mcon_collection_free(collection);
return con;
}
PHP_METHOD(MongoCollection, __construct) {
zval *parent, *name, *zns, *w, *wtimeout;
mongo_collection *c;
mongo_db *db;
char *ns, *name_str;
int name_len;
if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "Os", &parent, mongo_ce_DB, &name_str, &name_len) == FAILURE) {
zval *object = getThis();
ZVAL_NULL(object);
return;
}
// check for empty collection name
if (name_len == 0) {
#if ZEND_MODULE_API_NO >= 20060613
zend_throw_exception_ex(zend_exception_get_default(TSRMLS_C), 0 TSRMLS_CC, "MongoDB::__construct(): invalid name %s", name_str);
#else
zend_throw_exception_ex(zend_exception_get_default(), 0 TSRMLS_CC, "MongoDB::__construct(): invalid name %s", name_str);
#endif /* ZEND_MODULE_API_NO >= 20060613 */
return;
}
c = (mongo_collection*)zend_object_store_get_object(getThis() TSRMLS_CC);
PHP_MONGO_GET_DB(parent);
c->link = db->link;
zval_add_ref(&db->link);
c->parent = parent;
zval_add_ref(&parent);
MAKE_STD_ZVAL(name);
ZVAL_STRINGL(name, name_str, name_len, 1);
c->name = name;
spprintf(&ns, 0, "%s.%s", Z_STRVAL_P(db->name), Z_STRVAL_P(name));
MAKE_STD_ZVAL(zns);
ZVAL_STRING(zns, ns, 0);
c->ns = zns;
mongo_read_preference_copy(&db->read_pref, &c->read_pref);
w = zend_read_property(mongo_ce_DB, parent, "w", strlen("w"), NOISY TSRMLS_CC);
zend_update_property_long(mongo_ce_Collection, getThis(), "w", strlen("w"), Z_LVAL_P(w) TSRMLS_CC);
wtimeout = zend_read_property(mongo_ce_DB, parent, "wtimeout", strlen("wtimeout"), NOISY TSRMLS_CC);
zend_update_property_long(mongo_ce_Collection, getThis(), "wtimeout", strlen("wtimeout"), Z_LVAL_P(wtimeout) TSRMLS_CC);
}
/* {{{ MongoDB::__construct
*/
PHP_METHOD(MongoDB, __construct) {
zval *zlink;
char *name;
int name_len;
mongo_db *db;
mongoclient *link;
if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "Os", &zlink, mongo_ce_MongoClient, &name, &name_len) == FAILURE) {
zval *object = getThis();
ZVAL_NULL(object);
return;
}
if (0 == name_len ||
0 != strchr(name, ' ') || 0 != strchr(name, '.') || 0 != strchr(name, '\\') ||
0 != strchr(name, '/') || 0 != strchr(name, '$')) {
#if ZEND_MODULE_API_NO >= 20060613
zend_throw_exception_ex(zend_exception_get_default(TSRMLS_C), 0 TSRMLS_CC, "MongoDB::__construct(): invalid name %s", name);
#else
zend_throw_exception_ex(zend_exception_get_default(), 0 TSRMLS_CC, "MongoDB::__construct(): invalid name %s", name);
#endif /* ZEND_MODULE_API_NO >= 20060613 */
return;
}
db = (mongo_db*)zend_object_store_get_object(getThis() TSRMLS_CC);
db->link = zlink;
zval_add_ref(&db->link);
PHP_MONGO_GET_LINK(zlink);
if (link->servers->options.default_w != -1) {
zend_update_property_long(mongo_ce_DB, getThis(), "w", strlen("w"), link->servers->options.default_w TSRMLS_CC);
} else if (link->servers->options.default_wstring != NULL) {
zend_update_property_string(mongo_ce_DB, getThis(), "w", strlen("w"), link->servers->options.default_wstring TSRMLS_CC);
}
if (link->servers->options.default_wtimeout != -1) {
zend_update_property_long(mongo_ce_DB, getThis(), "wtimeout", strlen("wtimeout"), link->servers->options.default_wtimeout TSRMLS_CC);
}
mongo_read_preference_copy(&link->servers->read_pref, &db->read_pref);
MAKE_STD_ZVAL(db->name);
ZVAL_STRING(db->name, name, 1);
}
void mongo_servers_copy(mongo_servers *to, mongo_servers *from, int flags)
{
int i;
to->count = from->count;
for (i = 0; i < from->count; i++) {
to->server[i] = calloc(1, sizeof(mongo_server_def));
mongo_server_def_copy(to->server[i], from->server[i], flags);
}
to->options.con_type = from->options.con_type;
if (from->options.repl_set_name) {
to->options.repl_set_name = strdup(from->options.repl_set_name);
}
if (from->options.gssapiServiceName) {
to->options.gssapiServiceName = strdup(from->options.gssapiServiceName);
}
to->options.connectTimeoutMS = from->options.connectTimeoutMS;
to->options.socketTimeoutMS = from->options.socketTimeoutMS;
to->options.secondaryAcceptableLatencyMS = from->options.secondaryAcceptableLatencyMS;
to->options.default_w = from->options.default_w;
to->options.default_wtimeout = from->options.default_wtimeout;
if (from->options.default_wstring) {
to->options.default_wstring = strdup(from->options.default_wstring);
}
to->options.default_fsync = from->options.default_fsync;
to->options.default_journal = from->options.default_journal;
to->options.ssl = from->options.ssl;
if (from->options.ctx) {
to->options.ctx = from->options.ctx;
}
mongo_read_preference_copy(&from->read_pref, &to->read_pref);
}
void mongo_read_preference_replace(mongo_read_preference *from, mongo_read_preference *to)
{
mongo_read_preference_dtor(to);
mongo_read_preference_copy(from, to);
}
