Add function to check if the server is MariaDB

mscdex · Aug 20, 2012 · 6e7084e · 6e7084e
1 parent 880b1c0
commit 6e7084e
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/lib/Client.js b/lib/Client.js
@@ -61,6 +61,10 @@ Client.prototype.isConnected = function() {
   return this._client.connected;
 };
 
+Client.prototype.isMariaDB = function() {
+  return this._client.isMariaDB();
+};
+
 Client.prototype._reset = function() {
   this._closeOnEmpty = false;
   this._queries = [];

diff --git a/src/client.cc b/src/client.cc
@@ -549,6 +549,17 @@ class Client : public ObjectWrap {
       return Undefined();
     }
 
+    static Handle<Value> IsMariaDB(const Arguments& args) {
+      HandleScope scope;
+      Client* obj = ObjectWrap::Unwrap<Client>(args.This());
+      if (obj->state < STATE_CONNECTED) {
+        return ThrowException(Exception::Error(
+          String::New("Not connected"))
+        );
+      }
+      return scope.Close(Boolean::New(mariadb_connection(&obj->mysql)));
+    }
+
     static void Initialize(Handle<Object> target) {
       HandleScope scope;
 
@@ -564,6 +575,7 @@ class Client : public ObjectWrap {
       NODE_SET_PROTOTYPE_METHOD(Client_constructor, "abortQuery", AbortQuery);
       NODE_SET_PROTOTYPE_METHOD(Client_constructor, "escape", Escape);
       NODE_SET_PROTOTYPE_METHOD(Client_constructor, "end", Close);
+      NODE_SET_PROTOTYPE_METHOD(Client_constructor, "isMariaDB", IsMariaDB);
 
       emit_symbol = NODE_PSYMBOL("emit");
       target->Set(name, Client_constructor->GetFunction());