Created
August 6, 2018 12:25
-
-
Save danesparza/728890557046f2083e56c899d4c98524 to your computer and use it in GitHub Desktop.
sqlx and ql testing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
package main | |
import ( | |
"database/sql" | |
"fmt" | |
"log" | |
_ "github.com/cznic/ql/driver" | |
"github.com/jmoiron/sqlx" | |
) | |
var personSchema = ` | |
CREATE TABLE person ( | |
first_name string, | |
last_name string, | |
email string | |
);` | |
var placeSchema = `CREATE TABLE place ( | |
country string, | |
city string, | |
telcode int | |
);` | |
// Person represents a single person | |
type Person struct { | |
FirstName string `db:"first_name" json:"first_name,omitempty"` | |
LastName string `db:"last_name" json:"last_name,omitempty"` | |
Email string `json:"email,omitempty"` | |
} | |
// Place represents a geographical location | |
type Place struct { | |
Country string | |
City sql.NullString | |
TelCode int | |
} | |
func main() { | |
// Disk file DB | |
db, err := sqlx.Connect("ql", "ql.db") | |
if err != nil { | |
log.Fatalln(err) | |
} | |
defer func() { | |
if err := db.Close(); err != nil { | |
fmt.Printf("There was an error closing the database: %s", err) | |
} | |
fmt.Println("OK") | |
}() | |
// exec the schema or fail; multi-statement Exec behavior varies between | |
// database drivers; pq will exec them all, sqlite3 won't, ymmv | |
tx := db.MustBegin() | |
tx.MustExec(personSchema) | |
tx.MustExec(placeSchema) | |
tx.Commit() | |
tx = db.MustBegin() | |
tx.MustExec("INSERT INTO person (first_name, last_name, email) VALUES ($1, $2, $3);", "Jason", "Moiron", "jmoiron@jmoiron.net") | |
tx.MustExec("INSERT INTO person (first_name, last_name, email) VALUES ($1, $2, $3);", "John", "Doe", "johndoeDNE@gmail.net") | |
tx.MustExec("INSERT INTO place (country, city, telcode) VALUES ($1, $2, $3);", "United States", "New York", 1) | |
tx.MustExec("INSERT INTO place (country, telcode) VALUES ($1, $2);", "Hong Kong", 852) | |
tx.MustExec("INSERT INTO place (country, telcode) VALUES ($1, $2);", "Singapore", 65) | |
// Named queries can use structs, so if you have an existing struct (i.e. person := &Person{}) that you have populated, you can pass it in as &person | |
tx.NamedExec("INSERT INTO person (first_name, last_name, email) VALUES (:first_name, :last_name, :email);", &Person{"Jane", "Citizen", "jane.citzen@example.com"}) | |
tx.Commit() | |
// Query the database, storing results in a []Person (wrapped in []interface{}) | |
people := []Person{} | |
db.Select(&people, "SELECT * FROM person ORDER BY first_name ASC;") | |
jason, john := people[0], people[1] | |
fmt.Printf("%#v\n%#v", jason, john) | |
// Person{FirstName:"Jason", LastName:"Moiron", Email:"jmoiron@jmoiron.net"} | |
// Person{FirstName:"John", LastName:"Doe", Email:"johndoeDNE@gmail.net"} | |
// You can also get a single result, a la QueryRow | |
jason = Person{} | |
err = db.Get(&jason, "SELECT * FROM person WHERE first_name=$1;", "Jason") | |
fmt.Printf("%#v\n", jason) | |
// Person{FirstName:"Jason", LastName:"Moiron", Email:"jmoiron@jmoiron.net"} | |
// if you have null fields and use SELECT *, you must use sql.Null* in your struct | |
places := []Place{} | |
err = db.Select(&places, "SELECT * FROM place ORDER BY telcode ASC;") | |
if err != nil { | |
fmt.Println(err) | |
return | |
} | |
usa, singsing, honkers := places[0], places[1], places[2] | |
fmt.Printf("%#v\n%#v\n%#v\n", usa, singsing, honkers) | |
// Place{Country:"United States", City:sql.NullString{String:"New York", Valid:true}, TelCode:1} | |
// Place{Country:"Singapore", City:sql.NullString{String:"", Valid:false}, TelCode:65} | |
// Place{Country:"Hong Kong", City:sql.NullString{String:"", Valid:false}, TelCode:852} | |
// Loop through rows using only one struct | |
place := Place{} | |
rows, err := db.Queryx("SELECT * FROM place;") | |
for rows.Next() { | |
err := rows.StructScan(&place) | |
if err != nil { | |
log.Fatalln(err) | |
} | |
fmt.Printf("%#v\n", place) | |
} | |
// Place{Country:"United States", City:sql.NullString{String:"New York", Valid:true}, TelCode:1} | |
// Place{Country:"Hong Kong", City:sql.NullString{String:"", Valid:false}, TelCode:852} | |
// Place{Country:"Singapore", City:sql.NullString{String:"", Valid:false}, TelCode:65} | |
// Named queries, using `:name` as the bindvar. Automatic bindvar support | |
// which takes into account the dbtype based on the driverName on sqlx.Open/Connect | |
_, err = db.NamedExec(`INSERT INTO person (first_name,last_name,email) VALUES (:first,:last,:email);`, | |
map[string]interface{}{ | |
"first": "Bin", | |
"last": "Smuth", | |
"email": "bensmith@allblacks.nz", | |
}) | |
// Selects Mr. Smith from the database | |
rows, err = db.NamedQuery(`SELECT * FROM person WHERE first_name=:fn;`, map[string]interface{}{"fn": "Bin"}) | |
// Named queries can also use structs. Their bind names follow the same rules | |
// as the name -> db mapping, so struct fields are lowercased and the `db` tag | |
// is taken into consideration. | |
rows, err = db.NamedQuery(`SELECT * FROM person WHERE first_name=:first_name;`, jason) | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment