Do not require database in connection

Gemfile

@@ -10,3 +10,4 @@ gem "rake"
 gem "rspec"
 gem "pry"
 gem "dotenv"
+gem "parallel"

Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    ruby_snowflake_client (1.1.0)
+    ruby_snowflake_client (1.3.0)
 
 GEM
   remote: https://rubygems.org/
@@ -10,6 +10,7 @@ GEM
     diff-lcs (1.5.0)
     dotenv (2.8.1)
     method_source (1.0.0)
+    parallel (1.23.0)
     pry (0.14.2)
       coderay (~> 1.1)
       method_source (~> 1.0)
@@ -34,6 +35,7 @@ PLATFORMS
 DEPENDENCIES
   bundler
   dotenv
+  parallel
   pry
   rake
   rspec

ext/client.go

@@ -21,16 +21,25 @@ import (
 )
 
 type SnowflakeClient struct {
-	db *sql.DB
+	db  *sql.DB
+	cfg *sf.Config
 }
 
-func (x SnowflakeClient) Fetch(statement C.VALUE) C.VALUE {
+func (x SnowflakeClient) Fetch(statement string) C.VALUE {
 	t1 := time.Now()
 
 	if LOG_LEVEL > 0 {
-		fmt.Println("statement", RbGoString(statement))
+		fmt.Println("statement", statement)
 	}
-	rows, err := x.db.QueryContext(sf.WithHigherPrecision(context.Background()), RbGoString(statement))
+	if LOG_LEVEL > 0 {
+		fmt.Println("getting conn")
+	}
+	dbCtx := context.Background()
+	conn, _ := x.db.Conn(dbCtx)
+	if LOG_LEVEL > 0 {
+		fmt.Println("got conn")
+	}
+	rows, err := conn.QueryContext(sf.WithHigherPrecision(context.Background()), statement)
 	if err != nil {
 		result := C.rb_class_new_instance(0, &empty, rbSnowflakeResultClass)
 		errStr := fmt.Sprintf("Query error: '%s'", err.Error())
@@ -42,22 +51,86 @@ func (x SnowflakeClient) Fetch(statement C.VALUE) C.VALUE {
 	if LOG_LEVEL > 0 {
 		fmt.Printf("Query duration: %s\n", time.Now().Sub(t1))
 	}
+
+	result := C.rb_class_new_instance(0, &empty, rbSnowflakeResultClass)
+	if LOG_LEVEL > 0 {
+		fmt.Println("create new instance")
+	}
+	cols, _ := rows.Columns()
+	for idx, col := range cols {
+		col := col
+		cols[idx] = strings.ToLower(col)
+	}
+	rs := SnowflakeResult{rows, cols, conn}
+	resultMap[result] = &rs
+	if LOG_LEVEL > 0 {
+		fmt.Println("after for & map")
+	}
+	C.rb_ivar_set(result, RESULT_DURATION, RbNumFromDouble(C.double(duration)))
+	if LOG_LEVEL > 0 {
+		fmt.Println("end of func")
+	}
+	return result
+}
+
+func (x SnowflakeClient) FetchWithDB(statement string, dbName string) C.VALUE {
+	t1 := time.Now()
+
+	if LOG_LEVEL > 0 {
+		fmt.Println("statement", statement)
+	}
+	if LOG_LEVEL > 0 {
+		fmt.Println("getting conn")
+	}
+	//dbCtx, _ := context.WithTimeout(context.Background(), 1*time.Second)
+	dbCtx := context.Background()
+	conn, _ := x.db.Conn(dbCtx)
+	if LOG_LEVEL > 0 {
+		fmt.Println("got conn")
+	}
+	stmt := fmt.Sprintf("USE DATABASE %s", dbName)
+	_, err := conn.ExecContext(context.Background(), stmt)
+	if err != nil {
+		result := C.rb_class_new_instance(0, &empty, rbSnowflakeResultClass)
+		errStr := fmt.Sprintf("Query %s had error \n error: '%s'", stmt, err.Error())
+		C.rb_ivar_set(result, ERROR_IDENT, RbString(errStr))
+		return result
+	}
+	if LOG_LEVEL > 0 {
+		fmt.Printf("exec duration: %s\n", time.Now().Sub(t1))
+		t1 = time.Now()
+	}
+	rows, err := conn.QueryContext(sf.WithHigherPrecision(context.Background()), statement)
 	if err != nil {
 		result := C.rb_class_new_instance(0, &empty, rbSnowflakeResultClass)
 		errStr := fmt.Sprintf("Query error: '%s'", err.Error())
 		C.rb_ivar_set(result, ERROR_IDENT, RbString(errStr))
 		return result
 	}
 
+	duration := time.Now().Sub(t1).Seconds()
+	if LOG_LEVEL > 0 {
+		fmt.Printf("Query duration: %s\n", time.Now().Sub(t1))
+	}
+
 	result := C.rb_class_new_instance(0, &empty, rbSnowflakeResultClass)
+	if LOG_LEVEL > 0 {
+		fmt.Println("create new instance")
+	}
 	cols, _ := rows.Columns()
 	for idx, col := range cols {
 		col := col
 		cols[idx] = strings.ToLower(col)
 	}
-	rs := SnowflakeResult{rows, cols}
+	rs := SnowflakeResult{rows, cols, conn}
 	resultMap[result] = &rs
+	if LOG_LEVEL > 0 {
+		fmt.Println("after for & map")
+	}
 	C.rb_ivar_set(result, RESULT_DURATION, RbNumFromDouble(C.double(duration)))
+	if LOG_LEVEL > 0 {
+		fmt.Println("end of func")
+	}
 	return result
 }
 
@@ -76,6 +149,7 @@ func Connect(self C.VALUE, account C.VALUE, warehouse C.VALUE, database C.VALUE,
 	}
 
 	dsn, err := sf.DSN(cfg)
+	sf.GetLogger().SetLogLevel("trace")
 	if err != nil {
 		errStr := fmt.Sprintf("Snowflake Config Creation Error: '%s'", err.Error())
 		C.rb_ivar_set(self, ERROR_IDENT, RbString(errStr))
@@ -86,13 +160,23 @@ func Connect(self C.VALUE, account C.VALUE, warehouse C.VALUE, database C.VALUE,
 		errStr := fmt.Sprintf("Connection Error: '%s'", err.Error())
 		C.rb_ivar_set(self, ERROR_IDENT, RbString(errStr))
 	}
-	rs := SnowflakeClient{db}
+	rs := SnowflakeClient{db, cfg}
 	clientRef[self] = &rs
 }
 
 //export ObjFetch
 func ObjFetch(self C.VALUE, statement C.VALUE) C.VALUE {
 	x, _ := clientRef[self]
+	stmt := RbGoString(statement)
+
+	return x.Fetch(stmt)
+}
+
+//export ObjFetchWithDB
+func ObjFetchWithDB(self C.VALUE, statement C.VALUE, database C.VALUE) C.VALUE {
+	x, _ := clientRef[self]
+	stmt := RbGoString(statement)
+	db := RbGoString(database)
 
-	return x.Fetch(statement)
+	return x.FetchWithDB(stmt, db)
 }

ext/result.go

@@ -20,6 +20,7 @@ import (
 type SnowflakeResult struct {
 	rows    *sql.Rows
 	columns []string
+	conn    *sql.Conn
 }
 
 func wrapRbRaise(err error) {
@@ -66,6 +67,7 @@ func GetRowsNoEnum(self C.VALUE) C.VALUE {
 		C.rb_ary_store(rbArr, C.long(idx), elem)
 	}
 
+	res.Close()
 	return rbArr
 }
 
@@ -95,6 +97,7 @@ func GetRows(self C.VALUE) C.VALUE {
 	if LOG_LEVEL > 0 {
 		fmt.Printf("done with rows.next: %s\n", time.Now().Sub(t1))
 	}
+	res.Close()
 
 	return self
 }
@@ -112,9 +115,18 @@ func ObjNextRow(self C.VALUE) C.VALUE {
 		r := res.ScanNextRow(false)
 		return r
 	}
+	res.Close()
 	return C.Qnil
 }
 
+func (res SnowflakeResult) Close() {
+	if LOG_LEVEL > 0 {
+		fmt.Println("called res.close")
+	}
+	res.rows.Close()
+	res.conn.Close()
+}
+
 func (res SnowflakeResult) ScanNextRow(debug bool) C.VALUE {
 	rows := res.rows
 	if LOG_LEVEL > 0 {

ext/ruby_snowflake.go

@@ -5,6 +5,7 @@ package main
 #include "ruby/ruby.h"
 void Connect(VALUE,VALUE,VALUE,VALUE,VALUE,VALUE,VALUE,VALUE);
 VALUE ObjFetch(VALUE,VALUE);
+VALUE ObjFetchWithDB(VALUE,VALUE,VALUE);
 VALUE ObjNextRow(VALUE);
 VALUE Inspect(VALUE);
 VALUE GetRows(VALUE);
@@ -66,6 +67,7 @@ func Init_ruby_snowflake_client_ext() {
 	C.rb_define_method(rbSnowflakeClientClass, C.CString("inspect"), (*[0]byte)(C.Inspect), 0)
 	C.rb_define_method(rbSnowflakeClientClass, C.CString("to_s"), (*[0]byte)(C.Inspect), 0)
 	C.rb_define_method(rbSnowflakeClientClass, C.CString("_fetch"), (*[0]byte)(C.ObjFetch), 1)
+	C.rb_define_method(rbSnowflakeClientClass, C.CString("_fetch_with_db"), (*[0]byte)(C.ObjFetchWithDB), 2)
 
 	if LOG_LEVEL > 0 {
 		fmt.Println("init ruby snowflake client")