Skip to content

Commit 337a1c7

Browse files
committed
creating in mem db
1 parent da558ed commit 337a1c7

File tree

4 files changed

+13
-9
lines changed

4 files changed

+13
-9
lines changed

Diff for: database.go

+6-2
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,6 @@ func InitEmptyDatabase() *sql.DB {
5050
fmt.Println("Error while connecting to database")
5151
panic(err)
5252
}
53-
log.Println("Database provider init")
5453
return db
5554

5655
}
@@ -87,7 +86,12 @@ func QueryForTwoColumns(db *sql.DB, query string) [][2]string {
8786
return props
8887
}
8988
func Populate(db *sql.DB, query string, inserts []string) {
90-
_, err := db.Exec(query)
89+
_, err := db.Exec("USE IMPOSTER")
90+
if err != nil {
91+
fmt.Println("Error while connecting to database")
92+
panic(err)
93+
}
94+
_, err = db.Exec(query)
9195
if err != nil {
9296
fmt.Println("Error while connecting to database")
9397
panic(err)

Diff for: inmemory-driver.go

+1-2
Original file line numberDiff line numberDiff line change
@@ -21,8 +21,7 @@ func (factory) Resolve(name string, options *driver.Options) (string, sql.Databa
2121

2222
func createTestDatabase() *memory.Database {
2323
const (
24-
dbName = "mydb"
25-
tableName = "mytable"
24+
dbName = "IMPOSTER"
2625
)
2726

2827
db := memory.NewDatabase(dbName)

Diff for: main.go

+4-3
Original file line numberDiff line numberDiff line change
@@ -55,8 +55,9 @@ func main() {
5555
for _, v := range inserts {
5656
log.Println(v)
5757
}
58-
var provider *sql.DB = InitEmptyDatabase()
59-
Populate(provider, createCommand, inserts)
58+
var memdn *sql.DB = InitEmptyDatabase()
59+
log.Println("Database provider init")
60+
Populate(memdn, createCommand, inserts)
6061

6162
// start proxying
6263
socket, err := net.Listen("tcp", "127.0.0.1:3307")
@@ -70,7 +71,7 @@ func main() {
7071
if err != nil {
7172
log.Fatalf("failed to accept connection: %s", err.Error())
7273
}
73-
go handleConn(originSocket, provider)
74+
go handleConn(originSocket, memdn)
7475
}
7576

7677
}

Diff for: sql-manipulator.go

+2-2
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ import (
88

99
func CreateSelectInsertionFromSchema(schemaName, tableName string, columns [][2]string) string {
1010
var sb strings.Builder
11-
write(&sb, `SELECT CONCAT('INSERT INTO ', '%s.%s ', 'SET ', `, schemaName, tableName)
11+
write(&sb, `SELECT CONCAT('INSERT INTO %s.%s SET ', `, "IMPOSTER" /*schschemaName*/, tableName)
1212
for i, v := range columns {
1313
name := v[0]
1414
_type := v[1]
@@ -17,7 +17,7 @@ func CreateSelectInsertionFromSchema(schemaName, tableName string, columns [][2]
1717
setVal := `x.%s`
1818
if nullVal != 0 {
1919
isNull = "CAST(x.%s AS CHAR)"
20-
setVal = "CONCAT('\"', x.%s, '\"')"
20+
setVal = "QUOTE(x.%s)"
2121
}
2222
isNull = fmt.Sprintf(isNull, name)
2323
setVal = fmt.Sprintf(setVal, name)

0 commit comments

Comments
 (0)