Adding support for connecting through unix sockets

Author: mariano

src/connection.cc

@@ -17,6 +17,14 @@ node_db_mysql::Connection::~Connection() {
     }
 }
 
+void node_db_mysql::Connection::setSocket(const std::string& socket) {
+    this->socket = socket;
+}
+
+std::string node_db_mysql::Connection::getSocket() const {
+    return this->socket;
+}
+
 void node_db_mysql::Connection::open() throw(node_db::Exception&) {
     this->close();
 
@@ -27,7 +35,7 @@ void node_db_mysql::Connection::open() throw(node_db::Exception&) {
         this->password.c_str(),
         this->database.c_str(),
         this->port,
-        NULL,
+        !this->socket.empty() ? this->socket.c_str() : NULL,
         0);
     if (!this->opened) {
         throw node_db::Exception(mysql_error(this->connection));

src/connection.h

@@ -12,12 +12,17 @@ class Connection : public node_db::Connection {
     public:
         Connection();
         ~Connection();
+        std::string getSocket() const;
+        void setSocket(const std::string& user);
         void open() throw(node_db::Exception&);
         void close();
         std::string escape(const std::string& string) const throw(node_db::Exception&);
         std::string version() const;
         node_db::Result* query(const std::string& query) const throw(node_db::Exception&);
 
+    protected:
+        std::string socket;
+
     private:
         MYSQL* connection;
 };

src/mysql.cc

@@ -48,6 +48,22 @@ v8::Handle<v8::Value> node_db_mysql::Mysql::New(const v8::Arguments& args) {
     return scope.Close(args.This());
 }
 
+v8::Handle<v8::Value> node_db_mysql::Mysql::set(const v8::Arguments& args) {
+    v8::Handle<v8::Value> result = node_db::Binding::set(args);
+
+    v8::Local<v8::Object> options = args[0]->ToObject();
+    ARG_CHECK_OBJECT_ATTR_OPTIONAL_STRING(options, socket);
+
+    node_db_mysql::Connection* connection = static_cast<node_db_mysql::Connection*>(this->connection);
+
+    if (options->Has(socket_key)) {
+        v8::String::Utf8Value socket(options->Get(socket_key)->ToString());
+        connection->setSocket(*socket);
+    }
+
+    return result;
+}
+
 v8::Persistent<v8::Object> node_db_mysql::Mysql::createQuery() const {
     v8::Persistent<v8::Object> query(
         node_db_mysql::Query::constructorTemplate->GetFunction()->NewInstance());

src/mysql.h

@@ -18,6 +18,7 @@ class Mysql : public node_db::Binding {
         Mysql();
         ~Mysql();
         static v8::Handle<v8::Value> New(const v8::Arguments& args);
+        v8::Handle<v8::Value> set(const v8::Arguments& args);
         v8::Persistent<v8::Object> createQuery() const;
 };
 }