Partially use plugin to replace builtin db library.

Author: st0012

model.gb

@@ -1,13 +1,16 @@
+require "db"
+require "plugin"
+
 db_user = ENV["POSTGRES_USER"] || "postgres"
 db_name = ENV["POSTGRES_DB"]   || "goby_test"
 db_host = ENV["POSTGRES_HOST"] || "0.0.0.0"
 
-require "plugin"
+PG = DB.open("postgres", String.fmt("user=%s dbname=%s host=%s sslmode=disable", db_user, db_name, db_host))
 
 plugin = Plugin.generate("db") do |p|
-  p.import_pkg("", "database/sql")
+  p.import_pkg("", "github.com/jmoiron/sqlx")
   p.import_pkg("_", "github.com/lib/pq")
-  p.link_function("sql", "Open")
+  p.link_function("sqlx", "Open")
 end
 
 conn, err = plugin.go_func("Open", "postgres", String.fmt("user=%s dbname=%s host=%s sslmode=disable", db_user, db_name, db_host))
@@ -17,9 +20,43 @@ if err
   return nil
 end
 
-PG = conn
+class PluginDB
+  def initialize(conn)
+    @conn = conn
+  end
+
+  def run(query, *args)
+    r, err = @conn.go_func("Exec", query, *args)
+
+    if err != nil
+      puts(err.go_func("Error"))
+    end
+  end
+
+  def exec(query, *args)
+    id = 0
+    err = @conn.go_func("QueryRow", String.fmt("%s RETURNING id", query), *args).go_func("Scan", id.ptr)
+
+    if err != nil
+      puts(err.go_func("Error"))
+    else
+      id
+    end
+  end
+
+#   def query(query, *args)
+#     rows, err = conn.go_func("Queryx", query, *args)
+
+#     while rows.go_func("Next") do
+
+#     end
+#   end
+end
+
+PluginPG = PluginDB.new(conn)
 
-PG.run("create table if not exists list_items (
+
+PluginPG.run("create table if not exists list_items (
   id      serial primary key,
   title   varchar(40),
   checked boolean
@@ -36,37 +73,38 @@ class ListItem
   end
 
   def check
-    ListItem.db.exec('UPDATE list_items SET checked = true WHERE id = $1', @id)
+    self.class.plugin_db.exec('UPDATE list_items SET checked = true WHERE id = $1', @id)
     @checked = true
   end
 
   def uncheck
-    ListItem.db.exec('UPDATE list_items SET checked = false WHERE id = $1', @id)
+    self.class.plugin_db.exec('UPDATE list_items SET checked = false WHERE id = $1', @id)
     @checked = false
   end
 
   def update_title(title)
-    ListItem.db.exec('UPDATE list_items SET title = $1 WHERE id = $2', title, @id)
+    self.class.plugin_db.exec('UPDATE list_items SET title = $1 WHERE id = $2', title, @id)
   end
 
   def destroy
-    ListItem.db.exec('DELETE FROM list_items WHERE id = $1', @id)
+    self.class.plugin_db.exec('DELETE FROM list_items WHERE id = $1', @id)
   end
 
   def valid?
     @error.nil?
   end
 
-  def self.db
-    PG
+
+  def self.plugin_db
+    PluginPG
   end
 
   def self.all
-    self.db.query("SELECT * FROM list_items ORDER BY id DESC")
+    PG.query("SELECT * FROM list_items ORDER BY id DESC")
   end
 
   def self.find(id)
-    result = self.db.query("SELECT * FROM list_items WHERE id = $1", id).first
+    result = PG.query("SELECT * FROM list_items WHERE id = $1", id).first
     if result
       new({ id: result[:id], title: result[:title], checked: result[:checked] })
     end
@@ -77,7 +115,7 @@ class ListItem
       if result[:error].nil?
         title   = params[:title]
         checked = params[:checked].to_i == 1
-        resultID = self.db.exec("INSERT INTO list_items (title, checked) VALUES ($1, $2)", title, checked)
+        resultID = self.plugin_db.exec("INSERT INTO list_items (title, checked) VALUES ($1, $2)", title, checked)
         new({ id: resultID, title: title, checked: checked })
       else
         new({ error: result[:error] })