WebSVN – HomeAutomation – Diff – /trunk/PcSoftware/Atom1.5/src/vm/plugin/MySql.cpp

   this->ExportFunction("MySqlExport_Connect",       MySql::Export_Connect);
   this->ExportFunction("MySqlExport_Close",         MySql::Export_Close);
   this->ExportFunction("MySqlExport_Query",         MySql::Export_Query);
   this->ExportFunction("MySqlExport_SelectDb",      MySql::Export_SelectDb);
   this->ExportFunction("MySqlExport_AffectedRows",  MySql::Export_AffectedRows);
+  this->ExportFunction("MySqlExport_InsertId",      MySql::Export_InsertId);
+}
 MySql::~MySql()
+{
+}
+}
 void MySql::CallOutput(unsigned int request_id, std::string output)
+{
   atom::log::Info(log_module_, output);
+}
+MySql::ResourceId MySql::GetFreeResourceId()
+{
+  ResourceId resource_id = 1;
+  while (NULL != MySql::resources_[resource_id])
+  {
+    resource_id++;
+  }
+  return resource_id;
+}
 Value MySql::Export_Connect(const v8::Arguments& args)
+{
   ATOM_VM_PLUGIN_SCOPE;
-  MYSQL* resource = NULL;
+  MYSQL*      resource = NULL;
+  ResourceId  resource_id;
-  atom::log::Debug(log_module_, "%s called!", __FUNCTION__);
+  //atom::log::Debug(log_module_, "%s called!", __FUNCTION__);
   try
+  {
     ATOM_VM_PLUGIN_NUM_PARAMS(3);
     /* Initialize MySQL resource */
     resource = mysql_init(NULL);
     if (NULL == resource)
+    {
-      throw atom::exception::initialization_failed;
+      throw atom::exception::initialization_failed();
+    }
     /* Connect to database */
     if (mysql_real_connect(resource, *server, *username, *password, NULL, 0, NULL, 0) == NULL)
+    {
       atom::log::Error(log_module_, "Failed to connect to database %s, error code %d, error message \"%s\".", *server, mysql_errno(resource), mysql_error(resource));
-      throw atom::exception::connect_failed;
+      throw atom::exception::connect_failed();
+    }
     /* Add resource to list */
+    resource_id = MySql::GetFreeResourceId();
-    MySql::resources_[(uintptr_t)resource] = resource;
+    MySql::resources_[resource_id] = resource;
-    return handle_scope.Close(v8::Uint32::New((uintptr_t)resource));
+    return handle_scope.Close(v8::Uint32::New(resource_id));
+  }
   catch (std::exception& exception)
+  {
     if (NULL != resource)
+    {
+  }
+}
 Value MySql::Export_Close(const v8::Arguments& args)
+{
   ATOM_VM_PLUGIN_SCOPE;
-  atom::log::Debug(log_module_, "%s called!", __FUNCTION__);
+  //atom::log::Debug(log_module_, "%s called!", __FUNCTION__);
   try
   {
     ATOM_VM_PLUGIN_NUM_PARAMS(1);
     /* Check that the resource exist */
-    if (NULL == MySql::resources_[(uintptr_t)args[0]->Uint32Value()])
+    if (NULL == MySql::resources_[(ResourceId)args[0]->Uint32Value()])
     {
-      throw atom::exception::missing_resource;
+      throw atom::exception::missing_resource();
     }
     /* Close connection */
-    mysql_close(MySql::resources_[(uintptr_t)args[0]->Uint32Value()]);
+    mysql_close(MySql::resources_[(ResourceId)args[0]->Uint32Value()]);
-    MySql::resources_.erase((uintptr_t)args[0]->Uint32Value());
+    MySql::resources_.erase((ResourceId)args[0]->Uint32Value());
     return handle_scope.Close(v8::Boolean::New(true));
+  }
   catch (std::exception& exception)
+  {
     atom::log::Exception(log_module_, exception);
     return handle_scope.Close(v8::Boolean::New(false));
+  }
 }
 Value MySql::Export_Query(const v8::Arguments& args)
+{
   ATOM_VM_PLUGIN_SCOPE;
   MYSQL_RES* result = NULL;
-  atom::log::Debug(log_module_, "%s called!", __FUNCTION__);
+  //atom::log::Debug(log_module_, "%s called!", __FUNCTION__);
   try
   {
     ArgumentListPointer result_array = ArgumentListPointer(new ArgumentList);
     ATOM_VM_PLUGIN_NUM_PARAMS(2);
     v8::String::AsciiValue query(args[1]);
     /* Check that the resource exist */
-    if (NULL == MySql::resources_[(uintptr_t)args[0]->Uint32Value()])
+    if (NULL == MySql::resources_[(ResourceId)args[0]->Uint32Value()])
+    {
+      throw atom::exception::missing_resource();
+    }
+    /* Execute the query */
+    if (0 != mysql_query(MySql::resources_[(ResourceId)args[0]->Uint32Value()], *query))
+    {
-      throw atom::exception::missing_resource;
-    }
-    /* Execute the query */
-    if (0 != mysql_query(MySql::resources_[(uintptr_t)args[0]->Uint32Value()], *query))
-    {
-      throw atom::exception::action_failed;
+      throw atom::exception::action_failed();
+    }
     /* Get the result */
-    result = mysql_store_result(MySql::resources_[(uintptr_t)args[0]->Uint32Value()]);
+    result = mysql_store_result(MySql::resources_[(ResourceId)args[0]->Uint32Value()]);
     if (NULL == result)
     {
-      if (mysql_field_count(MySql::resources_[(uintptr_t)args[0]->Uint32Value()]) > 0)
+      if (mysql_field_count(MySql::resources_[(ResourceId)args[0]->Uint32Value()]) > 0)
       {
-        throw atom::exception::action_failed;
+        throw atom::exception::action_failed();
       }
     }
     else
+    {
       MYSQL_ROW           row;
       uint32_t            number_of_fields = mysql_num_fields(result);
       MYSQL_FIELD*        fields = mysql_fetch_fields(result);
       uint32_t            row_index = 0;
       v8::Local<v8::Array> result_array = v8::Array::New();
       while (NULL != (row = mysql_fetch_row(result)))
+      {
         v8::Local<v8::Object> vars = v8::Object::New();
 //          atom::log::Info(log_module_, "%s = %s\n", fields[index].name, row[index]);
+        }
         result_array->Set(row_index, vars);
         row_index++;
       }
       mysql_free_result(result);
       result = NULL;
       return handle_scope.Close(v8::Handle<v8::Value>(result_array));
     }
     return handle_scope.Close(v8::Boolean::New(true));
   }
   catch (std::exception& exception)
   {
     if (NULL != result)
+    {
       mysql_free_result(result);
       result = NULL;
     }
     atom::log::Exception(log_module_, exception);
     return handle_scope.Close(v8::Boolean::New(false));
+  }
 }
 Value MySql::Export_SelectDb(const v8::Arguments& args)
+{
   ATOM_VM_PLUGIN_SCOPE;
-  atom::log::Debug(log_module_, "%s called!", __FUNCTION__);
+  //atom::log::Debug(log_module_, "%s called!", __FUNCTION__);
   try
   {
     ATOM_VM_PLUGIN_NUM_PARAMS(2);
     v8::String::AsciiValue database(args[1]);
     /* Check that the resource exist */
-    if (NULL == MySql::resources_[(uintptr_t)args[0]->Uint32Value()])
+    if (NULL == MySql::resources_[(ResourceId)args[0]->Uint32Value()])
     {
-      throw atom::exception::missing_resource;
+      throw atom::exception::missing_resource();
     }
     /* Select database */
-    if (0 != mysql_select_db(MySql::resources_[(uintptr_t)args[0]->Uint32Value()], *database))
+    if (0 != mysql_select_db(MySql::resources_[(ResourceId)args[0]->Uint32Value()], *database))
+    {
-       throw atom::exception::action_failed;
+       throw atom::exception::action_failed();
+    }
     return handle_scope.Close(v8::Boolean::New(true));
   }
   catch (std::exception& exception)
   {
     atom::log::Exception(log_module_, exception);
     return handle_scope.Close(v8::Boolean::New(false));
   }
 }
 Value MySql::Export_AffectedRows(const v8::Arguments& args)
+{
+  ATOM_VM_PLUGIN_SCOPE;
+  //atom::log::Debug(log_module_, "%s called!", __FUNCTION__);
+  try
+  {
+    ATOM_VM_PLUGIN_NUM_PARAMS(1);
+    /* Check that the resource exist */
+    if (NULL == MySql::resources_[(ResourceId)args[0]->Uint32Value()])
+    {
+      throw atom::exception::missing_resource();
+    }
+    /* Select database */
+    uint32_t number_of_rows = mysql_affected_rows(MySql::resources_[(ResourceId)args[0]->Uint32Value()]);
+    return handle_scope.Close(v8::Uint32::New(number_of_rows));
+  }
+  catch (std::exception& exception)
+  {
+    atom::log::Exception(log_module_, exception);
+    return handle_scope.Close(v8::Boolean::New(false));
+  }
+}
+Value MySql::Export_InsertId(const v8::Arguments& args)
+{
   ATOM_VM_PLUGIN_SCOPE;
-  atom::log::Debug(log_module_, "%s called!", __FUNCTION__);
+  //atom::log::Debug(log_module_, "%s called!", __FUNCTION__);
   try
+  {
     ATOM_VM_PLUGIN_NUM_PARAMS(1);
     /* Check that the resource exist */
-    if (NULL == MySql::resources_[(uintptr_t)args[0]->Uint32Value()])
+    if (NULL == MySql::resources_[(ResourceId)args[0]->Uint32Value()])
     {
-      throw atom::exception::missing_resource;
+      throw atom::exception::missing_resource();
     }
     /* Select database */
-    uint32_t number_of_rows = mysql_affected_rows(MySql::resources_[(uintptr_t)args[0]->Uint32Value()]);
+    uint32_t insert_id = mysql_insert_id(MySql::resources_[(ResourceId)args[0]->Uint32Value()]);
-    return handle_scope.Close(v8::Uint32::New(number_of_rows));
+    return handle_scope.Close(v8::Uint32::New(insert_id));
+  }
   catch (std::exception& exception)
+  {
     atom::log::Exception(log_module_, exception);
     return handle_scope.Close(v8::Boolean::New(false));
+  }
+}
 }; // namespace plugin
 }; // namespace vm
 }; // namespace atom

Subversion Repositories HomeAutomation

(root)/trunk/PcSoftware/Atom1.5/src/vm/plugin/MySql.cpp – Rev 1945 → 1947