fork github.com/doug-martin

2025-03-22 23:02:05 +08:00
commit f14642a736
131 changed files with 34555 additions and 0 deletions
@@ -0,0 +1,255 @@
+package db_test
+
+import (
+	"fmt"
+	"time"
+
+	dbv2 "git.fsdpf.net/go/db/v2"
+	_ "git.fsdpf.net/go/db/v2/dialect/mysql"
+	_ "git.fsdpf.net/go/db/v2/dialect/postgres"
+	_ "git.fsdpf.net/go/db/v2/dialect/sqlite3"
+	"github.com/DATA-DOG/go-sqlmock"
+)
+
+// Creating a mysql dataset. Be sure to import the mysql adapter.
+func ExampleDialect_datasetMysql() {
+	// import _ "git.fsdpf.net/go/db/v2/dialect/mysql"
+
+	d := dbv2.Dialect("mysql")
+	ds := d.From("test").Where(dbv2.Ex{
+		"foo": "bar",
+		"baz": []int64{1, 2, 3},
+	}).Limit(10)
+
+	sql, args, _ := ds.ToSQL()
+	fmt.Println(sql, args)
+
+	sql, args, _ = ds.Prepared(true).ToSQL()
+	fmt.Println(sql, args)
+
+	// Output:
+	// SELECT * FROM `test` WHERE ((`baz` IN (1, 2, 3)) AND (`foo` = 'bar')) LIMIT 10 []
+	// SELECT * FROM `test` WHERE ((`baz` IN (?, ?, ?)) AND (`foo` = ?)) LIMIT ? [1 2 3 bar 10]
+}
+
+// Creating a mysql database. Be sure to import the mysql adapter.
+func ExampleDialect_dbMysql() {
+	// import _ "git.fsdpf.net/go/db/v2/dialect/mysql"
+
+	type item struct {
+		ID      int64  `db:"id"`
+		Address string `db:"address"`
+		Name    string `db:"name"`
+	}
+
+	// set up a mock db this would normally be
+	// db, err := sql.Open("mysql", dbURI)
+	// 	if err != nil {
+	// 		panic(err.Error())
+	// 	}
+	mDB, mock, _ := sqlmock.New()
+
+	d := dbv2.Dialect("mysql")
+
+	db := d.DB(mDB)
+
+	// use the db.From to get a dataset to execute queries
+	ds := db.From("items").Where(dbv2.C("id").Eq(1))
+
+	// set up mock for example purposes
+	mock.ExpectQuery("SELECT `address`, `id`, `name` FROM `items` WHERE \\(`id` = 1\\) LIMIT 1").
+		WillReturnRows(
+			sqlmock.NewRows([]string{"id", "address", "name"}).
+				FromCSVString("1, 111 Test Addr,Test1"),
+		)
+	var it item
+	found, err := ds.ScanStruct(&it)
+	fmt.Println(it, found, err)
+
+	// set up mock for example purposes
+	mock.ExpectQuery("SELECT `address`, `id`, `name` FROM `items` WHERE \\(`id` = \\?\\) LIMIT \\?").
+		WithArgs(1, 1).
+		WillReturnRows(
+			sqlmock.NewRows([]string{"id", "address", "name"}).
+				FromCSVString("1, 111 Test Addr,Test1"),
+		)
+
+	found, err = ds.Prepared(true).ScanStruct(&it)
+	fmt.Println(it, found, err)
+
+	// Output:
+	// {1 111 Test Addr Test1} true <nil>
+	// {1 111 Test Addr Test1} true <nil>
+}
+
+// Creating a mysql dataset. Be sure to import the postgres adapter
+func ExampleDialect_datasetPostgres() {
+	// import _ "git.fsdpf.net/go/db/v2/dialect/postgres"
+
+	d := dbv2.Dialect("postgres")
+	ds := d.From("test").Where(dbv2.Ex{
+		"foo": "bar",
+		"baz": []int64{1, 2, 3},
+	}).Limit(10)
+
+	sql, args, _ := ds.ToSQL()
+	fmt.Println(sql, args)
+
+	sql, args, _ = ds.Prepared(true).ToSQL()
+	fmt.Println(sql, args)
+
+	// Output:
+	// SELECT * FROM "test" WHERE (("baz" IN (1, 2, 3)) AND ("foo" = 'bar')) LIMIT 10 []
+	// SELECT * FROM "test" WHERE (("baz" IN ($1, $2, $3)) AND ("foo" = $4)) LIMIT $5 [1 2 3 bar 10]
+}
+
+// Creating a postgres dataset. Be sure to import the postgres adapter
+func ExampleDialect_dbPostgres() {
+	// import _ "git.fsdpf.net/go/db/v2/dialect/postgres"
+
+	type item struct {
+		ID      int64  `db:"id"`
+		Address string `db:"address"`
+		Name    string `db:"name"`
+	}
+
+	// set up a mock db this would normally be
+	// db, err := sql.Open("postgres", dbURI)
+	// 	if err != nil {
+	// 		panic(err.Error())
+	// 	}
+	mDB, mock, _ := sqlmock.New()
+
+	d := dbv2.Dialect("postgres")
+
+	db := d.DB(mDB)
+
+	// use the db.From to get a dataset to execute queries
+	ds := db.From("items").Where(dbv2.C("id").Eq(1))
+
+	// set up mock for example purposes
+	mock.ExpectQuery(`SELECT "address", "id", "name" FROM "items" WHERE \("id" = 1\) LIMIT 1`).
+		WillReturnRows(
+			sqlmock.NewRows([]string{"id", "address", "name"}).
+				FromCSVString("1, 111 Test Addr,Test1"),
+		)
+	var it item
+	found, err := ds.ScanStruct(&it)
+	fmt.Println(it, found, err)
+
+	// set up mock for example purposes
+	mock.ExpectQuery(`SELECT "address", "id", "name" FROM "items" WHERE \("id" = \$1\) LIMIT \$2`).
+		WithArgs(1, 1).
+		WillReturnRows(
+			sqlmock.NewRows([]string{"id", "address", "name"}).
+				FromCSVString("1, 111 Test Addr,Test1"),
+		)
+
+	found, err = ds.Prepared(true).ScanStruct(&it)
+	fmt.Println(it, found, err)
+
+	// Output:
+	// {1 111 Test Addr Test1} true <nil>
+	// {1 111 Test Addr Test1} true <nil>
+}
+
+// Creating a mysql dataset. Be sure to import the sqlite3 adapter
+func ExampleDialect_datasetSqlite3() {
+	// import _ "git.fsdpf.net/go/db/v2/dialect/sqlite3"
+
+	d := dbv2.Dialect("sqlite3")
+	ds := d.From("test").Where(dbv2.Ex{
+		"foo": "bar",
+		"baz": []int64{1, 2, 3},
+	}).Limit(10)
+
+	sql, args, _ := ds.ToSQL()
+	fmt.Println(sql, args)
+
+	sql, args, _ = ds.Prepared(true).ToSQL()
+	fmt.Println(sql, args)
+
+	// Output:
+	// SELECT * FROM `test` WHERE ((`baz` IN (1, 2, 3)) AND (`foo` = 'bar')) LIMIT 10 []
+	// SELECT * FROM `test` WHERE ((`baz` IN (?, ?, ?)) AND (`foo` = ?)) LIMIT ? [1 2 3 bar 10]
+}
+
+// Creating a sqlite3 database. Be sure to import the sqlite3 adapter
+func ExampleDialect_dbSqlite3() {
+	// import _ "git.fsdpf.net/go/db/v2/dialect/sqlite3"
+	type item struct {
+		ID      int64  `db:"id"`
+		Address string `db:"address"`
+		Name    string `db:"name"`
+	}
+
+	// set up a mock db this would normally be
+	// db, err := sql.Open("sqlite3", dbURI)
+	// 	if err != nil {
+	// 		panic(err.Error())
+	// 	}
+	mDB, mock, _ := sqlmock.New()
+
+	d := dbv2.Dialect("sqlite3")
+
+	db := d.DB(mDB)
+
+	// use the db.From to get a dataset to execute queries
+	ds := db.From("items").Where(dbv2.C("id").Eq(1))
+
+	// set up mock for example purposes
+	mock.ExpectQuery("SELECT `address`, `id`, `name` FROM `items` WHERE \\(`id` = 1\\) LIMIT 1").
+		WillReturnRows(
+			sqlmock.NewRows([]string{"id", "address", "name"}).
+				FromCSVString("1, 111 Test Addr,Test1"),
+		)
+	var it item
+	found, err := ds.ScanStruct(&it)
+	fmt.Println(it, found, err)
+
+	// set up mock for example purposes
+	mock.ExpectQuery("SELECT `address`, `id`, `name` FROM `items` WHERE \\(`id` = \\?\\) LIMIT \\?").
+		WithArgs(1, 1).
+		WillReturnRows(
+			sqlmock.NewRows([]string{"id", "address", "name"}).
+				FromCSVString("1, 111 Test Addr,Test1"),
+		)
+
+	found, err = ds.Prepared(true).ScanStruct(&it)
+	fmt.Println(it, found, err)
+
+	// Output:
+	// {1 111 Test Addr Test1} true <nil>
+	// {1 111 Test Addr Test1} true <nil>
+}
+
+func ExampleSetTimeLocation() {
+	loc, err := time.LoadLocation("Asia/Shanghai")
+	if err != nil {
+		panic(err)
+	}
+
+	created, err := time.Parse(time.RFC3339, "2019-10-01T15:01:00Z")
+	if err != nil {
+		panic(err)
+	}
+
+	// use original time with tz info
+	dbv2.SetTimeLocation(loc)
+	ds := dbv2.Insert("test").Rows(dbv2.Record{
+		"address": "111 Address",
+		"name":    "Bob Yukon",
+		"created": created,
+	})
+	sql, _, _ := ds.ToSQL()
+	fmt.Println(sql)
+
+	// convert time to UTC
+	dbv2.SetTimeLocation(time.UTC)
+	sql, _, _ = ds.ToSQL()
+	fmt.Println(sql)
+
+	// Output:
+	// INSERT INTO "test" ("address", "created", "name") VALUES ('111 Address', '2019-10-01T23:01:00+08:00', 'Bob Yukon')
+	// INSERT INTO "test" ("address", "created", "name") VALUES ('111 Address', '2019-10-01T15:01:00Z', 'Bob Yukon')
+}