Skip to content

Commit

Permalink
Merge pull request #4 from twharmon/no-globals
Browse files Browse the repository at this point in the history
remove some global variables
twharmon authored Jan 16, 2020
2 parents a5edb14 + d89dc1f commit b873252
Showing 7 changed files with 175 additions and 197 deletions.
115 changes: 85 additions & 30 deletions db.go
Original file line number Diff line number Diff line change
@@ -2,43 +2,89 @@ package gosql

import (
"database/sql"
"fmt"
"reflect"
"regexp"
"strings"
)

// DB is a wrapper around sql.DB.
type DB struct {
db *sql.DB
}

// SetMaxOpenConns sets the maximum number of open connections to the database.
//
// If MaxIdleConns is greater than 0 and the new MaxOpenConns is less than
// MaxIdleConns, then MaxIdleConns will be reduced to match the new
// MaxOpenConns limit.
//
// If n <= 0, then there is no limit on the number of open connections.
// The default is 0 (unlimited).
func (db *DB) SetMaxOpenConns(max int) {
db.db.SetMaxOpenConns(max)
}

// SetMaxIdleConns sets the maximum number of connections in the idle
// connection pool.
//
// If MaxOpenConns is greater than 0 but less than the new MaxIdleConns,
// then the new MaxIdleConns will be reduced to match the MaxOpenConns limit.
//
// If n <= 0, no idle connections are retained.
//
// The default max idle connections is currently 2. This may change in
// a future release.
func (db *DB) SetMaxIdleConns(max int) {
db.db.SetMaxIdleConns(max)
db *sql.DB
models map[string]*model
}

// Register .
func (db *DB) Register(structs ...interface{}) error {
for _, s := range structs {
if err := db.register(s); err != nil {
return err
}
}
return nil
}

func (db *DB) register(s interface{}) error {
typ := reflect.TypeOf(s)
if typ.Kind() == reflect.Ptr {
typ = typ.Elem()
}
if typ.Kind() != reflect.Struct {
return fmt.Errorf("you can only register structs, %s found", reflect.TypeOf(s).Kind())
}
m := new(model)
m.typ = typ
m.name = m.typ.Name()
m.table = toSnakeCase(m.name)
m.primaryFieldIndex = -1
for i := 0; i < m.typ.NumField(); i++ {
f := m.typ.Field(i)
tag, ok := f.Tag.Lookup("gosql")
if ok && tag == "-" {
continue
}
if ok && tag == "primary" {
m.primaryFieldIndex = i
}
m.fields = append(m.fields, toSnakeCase(f.Name))
}
if err := db.mustBeValid(m); err != nil {
return err
}
m.fieldCount = len(m.fields)
db.models[m.name] = m
return nil
}

func (db *DB) getModelOf(obj interface{}) (*model, error) {
t := reflect.TypeOf(obj)
if t.Kind() != reflect.Ptr {
return nil, fmt.Errorf("obj must be a pointer to your model struct")
}
t = t.Elem()
if t.Kind() != reflect.Struct {
return nil, fmt.Errorf("obj must be a pointer to your model struct")
}
m := db.models[t.Name()]
if m == nil {
return nil, fmt.Errorf("you must first register %s", t.Name())
}
return m, nil
}

func (db *DB) mustBeValid(m *model) error {
if db.models[m.name] != nil {
return fmt.Errorf("model %s found more than once", m.name)
}
if m.primaryFieldIndex < 0 {
return fmt.Errorf("model %s must have one and only one field tagged `gosql:\"primary\"`", m.name)
}
return nil
}

// Insert .
func (db *DB) Insert(obj interface{}) (sql.Result, error) {
m, err := getModelOf(obj)
m, err := db.getModelOf(obj)
if err != nil {
return nil, err
}
@@ -48,7 +94,7 @@ func (db *DB) Insert(obj interface{}) (sql.Result, error) {

// Update .
func (db *DB) Update(obj interface{}) (sql.Result, error) {
m, err := getModelOf(obj)
m, err := db.getModelOf(obj)
if err != nil {
return nil, err
}
@@ -58,7 +104,7 @@ func (db *DB) Update(obj interface{}) (sql.Result, error) {

// Delete .
func (db *DB) Delete(obj interface{}) (sql.Result, error) {
m, err := getModelOf(obj)
m, err := db.getModelOf(obj)
if err != nil {
return nil, err
}
@@ -113,3 +159,12 @@ func (db *DB) ManualDelete(table string) *DeleteQuery {
dq.table = table
return dq
}

var matchFirstCap = regexp.MustCompile("(.)([A-Z][a-z]+)")
var matchAllCap = regexp.MustCompile("([a-z0-9])([A-Z])")

func toSnakeCase(str string) string {
snake := matchFirstCap.ReplaceAllString(str, "${1}_${2}")
snake = matchAllCap.ReplaceAllString(snake, "${1}_${2}")
return strings.ToLower(snake)
}
96 changes: 55 additions & 41 deletions db_test.go
Original file line number Diff line number Diff line change
@@ -11,71 +11,60 @@ import (
"github.com/twharmon/gosql"
)

type User struct {
ID int `gosql:"primary"`
Name string
}

func init() {
if err := gosql.Register(User{}); err != nil {
panic(err)
}
}

func TestDelete(t *testing.T) {
type DeleteModel struct {
ID int `gosql:"primary"`
}
check(t, gosql.Register(DeleteModel{}))
deleteModel := DeleteModel{5}
db, mock, err := getMockDB()
check(t, err)
mock.ExpectExec(`^delete from delete_model where id = \?$`).WithArgs(deleteModel.ID).WillReturnResult(sqlmock.NewResult(0, 1))
type T struct {
ID int `gosql:"primary"`
}
check(t, db.Register(T{}))
deleteModel := T{5}
mock.ExpectExec(`^delete from t where id = \?$`).WithArgs(deleteModel.ID).WillReturnResult(sqlmock.NewResult(0, 1))
_, err = db.Delete(&deleteModel)
check(t, err)
check(t, mock.ExpectationsWereMet())
}

func TestUpdate(t *testing.T) {
type UpdateModel struct {
db, mock, err := getMockDB()
check(t, err)
type T struct {
ID int `gosql:"primary"`
Name string
}
check(t, gosql.Register(UpdateModel{}))
updateModel := UpdateModel{5, "foo"}
db, mock, err := getMockDB()
check(t, err)
mock.ExpectExec(`^update update_model set name = \? where id = \?$`).WithArgs(updateModel.Name, updateModel.ID).WillReturnResult(sqlmock.NewResult(0, 1))
check(t, db.Register(T{}))
updateModel := T{5, "foo"}
mock.ExpectExec(`^update t set name = \? where id = \?$`).WithArgs(updateModel.Name, updateModel.ID).WillReturnResult(sqlmock.NewResult(0, 1))
_, err = db.Update(&updateModel)
check(t, err)
check(t, mock.ExpectationsWereMet())
}

func TestInsert(t *testing.T) {
type InsertModel struct {
db, mock, err := getMockDB()
check(t, err)
type T struct {
ID int `gosql:"primary"`
Name string
}
check(t, gosql.Register(InsertModel{}))
insertModel := InsertModel{Name: "foo"}
db, mock, err := getMockDB()
check(t, err)
mock.ExpectExec(`^insert into insert_model \(name\) values \(\?\)$`).WithArgs(insertModel.Name).WillReturnResult(sqlmock.NewResult(0, 1))
check(t, db.Register(T{}))
insertModel := T{Name: "foo"}
mock.ExpectExec(`^insert into t \(name\) values \(\?\)$`).WithArgs(insertModel.Name).WillReturnResult(sqlmock.NewResult(0, 1))
_, err = db.Insert(&insertModel)
check(t, err)
check(t, mock.ExpectationsWereMet())
}

func TestInsertWithPrimary(t *testing.T) {
type InsertWithPrimaryModel struct {
db, mock, err := getMockDB()
check(t, err)
type T struct {
ID int `gosql:"primary"`
Name string
}
check(t, gosql.Register(InsertWithPrimaryModel{}))
insertModelWithPrimary := InsertWithPrimaryModel{5, "foo"}
db, mock, err := getMockDB()
check(t, err)
mock.ExpectExec(`^insert into insert_with_primary_model \(id, name\) values \(\?, \?\)$`).WithArgs(insertModelWithPrimary.ID, insertModelWithPrimary.Name).WillReturnResult(sqlmock.NewResult(0, 1))
check(t, db.Register(T{}))
insertModelWithPrimary := T{5, "foo"}
mock.ExpectExec(`^insert into t \(id, name\) values \(\?, \?\)$`).WithArgs(insertModelWithPrimary.ID, insertModelWithPrimary.Name).WillReturnResult(sqlmock.NewResult(0, 1))
_, err = db.Insert(&insertModelWithPrimary)
check(t, err)
check(t, mock.ExpectationsWereMet())
@@ -85,12 +74,12 @@ func ExampleDB_Insert() {
os.Remove("/tmp/foo.db")
sqliteDB, _ := sql.Open("sqlite3", "/tmp/foo.db")
sqliteDB.Exec("create table user (id integer not null primary key, name text); delete from user")
db := gosql.Conn(sqliteDB)
db := gosql.New(sqliteDB)
type User struct {
ID int `gosql:"primary"`
Name string
}
gosql.Register(User{})
db.Register(User{})
db.Insert(&User{Name: "Gopher"})
var user User
db.Select("*").To(&user)
@@ -102,12 +91,12 @@ func ExampleDB_Update() {
os.Remove("/tmp/foo.db")
sqliteDB, _ := sql.Open("sqlite3", "/tmp/foo.db")
sqliteDB.Exec("create table user (id integer not null primary key, name text); delete from user")
db := gosql.Conn(sqliteDB)
db := gosql.New(sqliteDB)
type User struct {
ID int `gosql:"primary"`
Name string
}
gosql.Register(User{})
db.Register(User{})
user := User{ID: 5, Name: "Gopher"}
db.Insert(&user)
user.Name = "Gofer"
@@ -122,12 +111,12 @@ func ExampleDB_Delete() {
os.Remove("/tmp/foo.db")
sqliteDB, _ := sql.Open("sqlite3", "/tmp/foo.db")
sqliteDB.Exec("create table user (id integer not null primary key, name text); delete from user")
db := gosql.Conn(sqliteDB)
db := gosql.New(sqliteDB)
type User struct {
ID int `gosql:"primary"`
Name string
}
gosql.Register(User{})
db.Register(User{})
user := User{ID: 5, Name: "Gopher"}
db.Insert(&user)
db.Delete(&user)
@@ -139,6 +128,11 @@ func ExampleDB_Delete() {

func BenchmarkInsert(b *testing.B) {
db := getSQLiteDB(b, "create table user (id integer not null primary key, name text); delete from user")
type User struct {
ID int `gosql:"primary"`
Name string
}
db.Register(User{})
user := User{Name: "Gopher"}
b.ResetTimer()
for i := 0; i < b.N; i++ {
@@ -149,6 +143,11 @@ func BenchmarkInsert(b *testing.B) {

func BenchmarkUpdate(b *testing.B) {
db := getSQLiteDB(b, "create table user (id integer not null primary key, name text); delete from user")
type User struct {
ID int `gosql:"primary"`
Name string
}
db.Register(User{})
user := User{Name: "Gopher"}
_, err := db.Insert(&user)
check(b, err)
@@ -161,6 +160,11 @@ func BenchmarkUpdate(b *testing.B) {

func BenchmarkSelect(b *testing.B) {
db := getSQLiteDB(b, "create table user (id integer not null primary key, name text); delete from user")
type User struct {
ID int `gosql:"primary"`
Name string
}
db.Register(User{})
user := User{ID: 5, Name: "Gopher"}
_, err := db.Insert(&user)
check(b, err)
@@ -173,6 +177,11 @@ func BenchmarkSelect(b *testing.B) {

func BenchmarkSelectMany(b *testing.B) {
db := getSQLiteDB(b, "create table user (id integer not null primary key, name text); delete from user")
type User struct {
ID int `gosql:"primary"`
Name string
}
db.Register(User{})
user := User{Name: "Gopher"}
for i := 0; i < 100; i++ {
_, err := db.Insert(&user)
@@ -187,6 +196,11 @@ func BenchmarkSelectMany(b *testing.B) {

func BenchmarkSelectManyPtrs(b *testing.B) {
db := getSQLiteDB(b, "create table user (id integer not null primary key, name text); delete from user")
type User struct {
ID int `gosql:"primary"`
Name string
}
db.Register(User{})
user := User{Name: "Gopher"}
for i := 0; i < 100; i++ {
_, err := db.Insert(&user)
Loading

0 comments on commit b873252

Please sign in to comment.