expose profiling api

lib/sqlite3.js

@@ -93,24 +93,24 @@ var isVerbose = false;
 
 Database.prototype.addListener = Database.prototype.on = function(type) {
     var val = EventEmitter.prototype.addListener.apply(this, arguments);
-    if (type === 'trace') {
-        this.configure('trace', true);
+    if (type === 'trace' || type === 'profile') {
+        this.configure(type, true);
     }
     return val;
 };
 
 Database.prototype.removeListener = function(type) {
     var val = EventEmitter.prototype.removeListener.apply(this, arguments);
-    if (type === 'trace' && !this._events[type]) {
-        this.configure('trace', false);
+    if ((type === 'trace' || type === 'profile') && !this._events[type]) {
+        this.configure(type, false);
     }
     return val;
 };
 
 Database.prototype.removeAllListeners = function(type) {
     var val = EventEmitter.prototype.removeAllListeners.apply(this, arguments);
-    if (type === 'trace') {
-        this.configure('trace', false);
+    if (type === 'trace' || type === 'profile') {
+        this.configure(type, false);
     }
     return val;
 };

src/database.cc

@@ -209,11 +209,7 @@ void Database::EIO_BeginClose(Baton* baton) {
     assert(baton->db->handle);
     assert(baton->db->pending == 0);
 
-    if (baton->db->debug_trace) {
-        delete baton->db->debug_trace;
-        baton->db->debug_trace = NULL;
-    }
-
+    baton->db->RemoveCallbacks();
     eio_custom(EIO_Close, EIO_PRI_DEFAULT, EIO_AfterClose, baton);
 }
 
@@ -315,6 +311,11 @@ Handle<Value> Database::Configure(const Arguments& args) {
         Baton* baton = new Baton(db, handle);
         db->Schedule(RegisterTraceCallback, baton);
     }
+    else if (args[0]->Equals(String::NewSymbol("profile"))) {
+        Local<Function> handle;
+        Baton* baton = new Baton(db, handle);
+        db->Schedule(RegisterProfileCallback, baton);
+    }
     else {
         return ThrowException(Exception::Error(String::Concat(
             args[0]->ToString(),
@@ -369,6 +370,49 @@ void Database::TraceCallback(EV_P_ ev_async *w, int revents) {
     queries.clear();
 }
 
+void Database::RegisterProfileCallback(Baton* baton) {
+    assert(baton->db->open);
+    assert(baton->db->handle);
+    Database* db = baton->db;
+
+    if (db->debug_profile == NULL) {
+        // Add it.
+        db->debug_profile = new AsyncProfile(db, ProfileCallback);
+        sqlite3_profile(db->handle, ProfileCallback, db);
+    }
+    else {
+        // Remove it.
+        sqlite3_profile(db->handle, NULL, NULL);
+        delete db->debug_profile;
+        db->debug_profile = NULL;
+    }
+
+    delete baton;
+}
+
+void Database::ProfileCallback(void* db, const char* sql, sqlite3_uint64 nsecs) {
+    // Note: This function is called in the thread pool.
+    // Note: Some queries, such as "EXPLAIN" queries, are not sent through this.
+    static_cast<Database*>(db)->debug_profile->send(ProfileInfo(sql, nsecs));
+}
+
+void Database::ProfileCallback(EV_P_ ev_async *w, int revents) {
+    // Note: This function is called in the main V8 thread.
+    HandleScope scope;
+    AsyncProfile* async = static_cast<AsyncProfile*>(w->data);
+
+    std::vector<ProfileInfo> queries = async->get();
+    for (unsigned int i = 0; i < queries.size(); i++) {
+        Local<Value> argv[] = {
+            String::NewSymbol("profile"),
+            String::New(queries[i].first.c_str()),
+            Integer::New((double)queries[i].second / 1000000.0)
+        };
+        EMIT_EVENT(async->parent->handle_, 3, argv);
+    }
+    queries.clear();
+}
+
 Handle<Value> Database::Exec(const Arguments& args) {
     HandleScope scope;
     Database* db = ObjectWrap::Unwrap<Database>(args.This());
@@ -511,6 +555,17 @@ int Database::EIO_AfterLoadExtension(eio_req *req) {
     return 0;
 }
 
+void Database::RemoveCallbacks() {
+    if (debug_trace) {
+        delete debug_trace;
+        debug_trace = NULL;
+    }
+    if (debug_profile) {
+        delete debug_profile;
+        debug_profile = NULL;
+    }
+}
+
 /**
  * Override this so that we can properly close the database when this object
  * gets garbage collected.
@@ -537,10 +592,7 @@ void Database::Unref() {
 void Database::Destruct(Persistent<Value> value, void *data) {
     Database* db = static_cast<Database*>(data);
 
-    if (db->debug_trace) {
-        delete db->debug_trace;
-        db->debug_trace = NULL;
-    }
+    db->RemoveCallbacks();
 
     if (db->handle) {
         eio_custom(EIO_Destruct, EIO_PRI_DEFAULT, EIO_AfterDestruct, db);

src/database.h

@@ -133,6 +133,8 @@ public:
     };
 
     typedef Async<std::string, Database> AsyncTrace;
+    typedef std::pair<std::string, sqlite3_uint64> ProfileInfo;
+    typedef Async<ProfileInfo, Database> AsyncProfile;
 
     friend class Statement;
 
@@ -143,7 +145,8 @@ protected:
         locked(false),
         pending(0),
         serialize(false),
-        debug_trace(NULL) {
+        debug_trace(NULL),
+        debug_profile(NULL) {
 
     }
 
@@ -185,7 +188,11 @@ protected:
     static void RegisterTraceCallback(Baton* baton);
     static void TraceCallback(void* db, const char* sql);
     static void TraceCallback(EV_P_ ev_async *w, int revents);
+    static void RegisterProfileCallback(Baton* baton);
+    static void ProfileCallback(void* db, const char* sql, sqlite3_uint64 nsecs);
+    static void ProfileCallback(EV_P_ ev_async *w, int revents);
 
+    void RemoveCallbacks();
     void Wrap (Handle<Object> handle);
     inline void MakeWeak();
     virtual void Unref();
@@ -205,6 +212,7 @@ protected:
     std::queue<Call*> queue;
 
     AsyncTrace* debug_trace;
+    AsyncProfile* debug_profile;
 };
 
 }