mongodb support
This commit is contained in:
Henrik Hautakoski
parent
97bdfd94f1
commit
a908cf92af
12 changed files with 2884 additions and 4 deletions
186
lib/mongodb/mongo.h
Normal file
186
lib/mongodb/mongo.h
Normal file
|
|
@ -0,0 +1,186 @@
|
|||
/* mongo.h */
|
||||
|
||||
/* Copyright 2009, 2010 10gen Inc.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef _MONGO_H_
|
||||
#define _MONGO_H_
|
||||
|
||||
#include "mongo_except.h"
|
||||
#include "bson.h"
|
||||
|
||||
#ifdef _WIN32
|
||||
#include <windows.h>
|
||||
#include <winsock.h>
|
||||
typedef int socklen_t;
|
||||
#else
|
||||
#include <arpa/inet.h>
|
||||
#include <sys/types.h>
|
||||
#include <sys/socket.h>
|
||||
#include <netinet/in.h>
|
||||
#include <netinet/tcp.h>
|
||||
#endif
|
||||
|
||||
MONGO_EXTERN_C_START
|
||||
|
||||
typedef struct mongo_connection_options {
|
||||
char host[255];
|
||||
int port;
|
||||
} mongo_connection_options;
|
||||
|
||||
typedef struct {
|
||||
mongo_connection_options* left_opts; /* always current server */
|
||||
mongo_connection_options* right_opts; /* unused with single server */
|
||||
struct sockaddr_in sa;
|
||||
socklen_t addressSize;
|
||||
int sock;
|
||||
bson_bool_t connected;
|
||||
mongo_exception_context exception;
|
||||
} mongo_connection;
|
||||
|
||||
#pragma pack(1)
|
||||
typedef struct {
|
||||
int len;
|
||||
int id;
|
||||
int responseTo;
|
||||
int op;
|
||||
} mongo_header;
|
||||
|
||||
typedef struct {
|
||||
mongo_header head;
|
||||
char data;
|
||||
} mongo_message;
|
||||
|
||||
typedef struct {
|
||||
int flag; /* non-zero on failure */
|
||||
int64_t cursorID;
|
||||
int start;
|
||||
int num;
|
||||
} mongo_reply_fields;
|
||||
|
||||
typedef struct {
|
||||
mongo_header head;
|
||||
mongo_reply_fields fields;
|
||||
char objs;
|
||||
} mongo_reply;
|
||||
#pragma pack()
|
||||
|
||||
typedef struct {
|
||||
mongo_reply * mm; /* message is owned by cursor */
|
||||
mongo_connection * conn; /* connection is *not* owned by cursor */
|
||||
const char* ns; /* owned by cursor */
|
||||
bson current;
|
||||
} mongo_cursor;
|
||||
|
||||
enum mongo_operations {
|
||||
mongo_op_msg = 1000, /* generic msg command followed by a string */
|
||||
mongo_op_update = 2001, /* update object */
|
||||
mongo_op_insert = 2002,
|
||||
mongo_op_query = 2004,
|
||||
mongo_op_get_more = 2005,
|
||||
mongo_op_delete = 2006,
|
||||
mongo_op_kill_cursors = 2007
|
||||
};
|
||||
|
||||
|
||||
/* ----------------------------
|
||||
CONNECTION STUFF
|
||||
------------------------------ */
|
||||
|
||||
typedef enum {
|
||||
mongo_conn_success = 0,
|
||||
mongo_conn_bad_arg,
|
||||
mongo_conn_no_socket,
|
||||
mongo_conn_fail,
|
||||
mongo_conn_not_master /* leaves conn connected to slave */
|
||||
} mongo_conn_return;
|
||||
|
||||
/**
|
||||
* @param options can be null
|
||||
*/
|
||||
mongo_conn_return mongo_connect( mongo_connection * conn , mongo_connection_options * options );
|
||||
mongo_conn_return mongo_connect_pair( mongo_connection * conn , mongo_connection_options * left, mongo_connection_options * right );
|
||||
mongo_conn_return mongo_reconnect( mongo_connection * conn ); /* you will need to reauthenticate after calling */
|
||||
bson_bool_t mongo_disconnect( mongo_connection * conn ); /* use this if you want to be able to reconnect */
|
||||
bson_bool_t mongo_destroy( mongo_connection * conn ); /* you must call this even if connection failed */
|
||||
|
||||
|
||||
|
||||
/* ----------------------------
|
||||
CORE METHODS - insert update remove query getmore
|
||||
------------------------------ */
|
||||
|
||||
void mongo_insert( mongo_connection * conn , const char * ns , bson * data );
|
||||
void mongo_insert_batch( mongo_connection * conn , const char * ns , bson ** data , int num );
|
||||
|
||||
static const int MONGO_UPDATE_UPSERT = 0x1;
|
||||
static const int MONGO_UPDATE_MULTI = 0x2;
|
||||
void mongo_update(mongo_connection* conn, const char* ns, const bson* cond, const bson* op, int flags);
|
||||
|
||||
void mongo_remove(mongo_connection* conn, const char* ns, const bson* cond);
|
||||
|
||||
mongo_cursor* mongo_find(mongo_connection* conn, const char* ns, bson* query, bson* fields ,int nToReturn ,int nToSkip, int options);
|
||||
bson_bool_t mongo_cursor_next(mongo_cursor* cursor);
|
||||
void mongo_cursor_destroy(mongo_cursor* cursor);
|
||||
|
||||
/* out can be NULL if you don't care about results. useful for commands */
|
||||
bson_bool_t mongo_find_one(mongo_connection* conn, const char* ns, bson* query, bson* fields, bson* out);
|
||||
|
||||
int64_t mongo_count(mongo_connection* conn, const char* db, const char* coll, bson* query);
|
||||
|
||||
/* ----------------------------
|
||||
HIGHER LEVEL - indexes - command helpers eval
|
||||
------------------------------ */
|
||||
|
||||
/* Returns true on success */
|
||||
/* WARNING: Unlike other drivers these do not cache results */
|
||||
|
||||
static const int MONGO_INDEX_UNIQUE = 0x1;
|
||||
static const int MONGO_INDEX_DROP_DUPS = 0x2;
|
||||
bson_bool_t mongo_create_index(mongo_connection * conn, const char * ns, bson * key, int options, bson * out);
|
||||
bson_bool_t mongo_create_simple_index(mongo_connection * conn, const char * ns, const char* field, int options, bson * out);
|
||||
|
||||
/* ----------------------------
|
||||
COMMANDS
|
||||
------------------------------ */
|
||||
|
||||
bson_bool_t mongo_run_command(mongo_connection * conn, const char * db, bson * command, bson * out);
|
||||
|
||||
/* for simple commands with a single k-v pair */
|
||||
bson_bool_t mongo_simple_int_command(mongo_connection * conn, const char * db, const char* cmd, int arg, bson * out);
|
||||
bson_bool_t mongo_simple_str_command(mongo_connection * conn, const char * db, const char* cmd, const char* arg, bson * out);
|
||||
|
||||
bson_bool_t mongo_cmd_drop_db(mongo_connection * conn, const char * db);
|
||||
bson_bool_t mongo_cmd_drop_collection(mongo_connection * conn, const char * db, const char * collection, bson * out);
|
||||
|
||||
void mongo_cmd_add_user(mongo_connection* conn, const char* db, const char* user, const char* pass);
|
||||
bson_bool_t mongo_cmd_authenticate(mongo_connection* conn, const char* db, const char* user, const char* pass);
|
||||
|
||||
/* return value is master status */
|
||||
bson_bool_t mongo_cmd_ismaster(mongo_connection * conn, bson * out);
|
||||
|
||||
/* true return indicates error */
|
||||
bson_bool_t mongo_cmd_get_last_error(mongo_connection * conn, const char * db, bson * out);
|
||||
bson_bool_t mongo_cmd_get_prev_error(mongo_connection * conn, const char * db, bson * out);
|
||||
void mongo_cmd_reset_error(mongo_connection * conn, const char * db);
|
||||
|
||||
/* ----------------------------
|
||||
UTILS
|
||||
------------------------------ */
|
||||
|
||||
MONGO_EXTERN_C_END
|
||||
|
||||
|
||||
#endif
|
||||
Reference in a new issue