apply xorm

Author: XieBiao

config/db.go

@@ -0,0 +1,36 @@
+package config
+
+import (
+	"github.com/BurntSushi/toml"
+	"errors"
+)
+
+type Connection struct {
+	Driver   string
+	Dsn      string
+	Log      bool
+	LogLevel int
+	LogFile  string
+}
+
+type Connections struct {
+	Connections map[string]Connection
+}
+
+func GetDbConfigs() map[string]Connection {
+
+	var conns Connections
+	if _, err := toml.DecodeFile("config/db.toml", &conns); err != nil {
+		panic(errors.New("parse db.toml fail"))
+	}
+	return conns.Connections
+}
+
+func GetDbConfig(id string) Connection {
+	conns := GetDbConfigs()
+	conn, err := conns[id]
+	if !err {
+		panic(errors.New("connection " + id + " is not available"))
+	}
+	return conn
+}

config/db.toml

@@ -0,0 +1,23 @@
+[connections]
+
+[connections.default]
+driver = "mysql"
+dsn = "root:123456@tcp(127.0.0.1:3306)/test"
+log = true
+loglevel = 0
+logfile = "./sql.log"
+
+[connections.db1]
+driver = "sqlite3"
+dsn = "./test.db"
+log = false
+loglevel = 0
+# empty logfile means log into console
+logfile = ""
+
+[connections.db2]
+driver = "mysql"
+dsn = "root:123456@tcp(127.0.0.1:3306)/db2"
+log = false
+loglevel = 0
+logfile = "./sql.log"

core/model/db.go

@@ -0,0 +1,41 @@
+package model
+
+import (
+	"github.com/go-xorm/xorm"
+	"github.com/hhxsv5/go-db-proxy-api/config"
+	"github.com/go-xorm/core"
+	"os"
+)
+
+type DB struct {
+	Cfg    config.Connection
+	Engine *xorm.Engine
+}
+
+func NewDB(conn string) *DB {
+	cfg := config.GetDbConfig(conn)
+	db := &DB{cfg, nil}
+
+	var err error
+	db.Engine, err = xorm.NewEngine(cfg.Driver, cfg.Dsn)
+	if err != nil {
+		panic(err)
+	}
+
+	if cfg.Log {
+		//log into file
+		if len(cfg.LogFile) > 0 {
+			f, err := os.OpenFile(cfg.LogFile, os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0666)
+			if err != nil {
+				panic(err)
+			}
+			db.Engine.SetLogger(xorm.NewSimpleLogger(f))
+		} //else log into console
+
+		db.Engine.ShowSQL(true)
+		db.Engine.ShowExecTime(true)
+		db.Engine.Logger().SetLevel(core.LogLevel(cfg.LogLevel))
+	}
+
+	return db
+}

drivers/base.go

@@ -1,20 +1,12 @@
 package main
 
 import (
-	"github.com/hhxsv5/go-db-proxy-api/drivers"
 	"fmt"
+	"github.com/hhxsv5/go-db-proxy-api/models"
 )
 
 func main() {
-	m := drivers.NewMySQL("user:password@tcp(120.0.0.1:3306)/dbname")
-	err := m.Open()
-	if err != nil {
-		panic(err.Error())
-	}
-	defer m.Close()
-
-	rows := m.Query("select id,cellphone from user order by id desc limit 5")
-
-	fmt.Println(rows)
-
+	user := models.CreateUser("18780207350")
+	fmt.Println(user)
+	fmt.Printf("%p", user)
 }

drivers/database.go

@@ -0,0 +1,23 @@
+package models
+
+import (
+	_ "github.com/go-sql-driver/mysql" //remember: import xxx driver to init
+	"github.com/go-xorm/xorm"
+	"github.com/hhxsv5/go-db-proxy-api/core/model"
+	"github.com/go-xorm/core"
+)
+
+const connection = "default" //default connection in file db.toml
+
+var (
+	Db  *model.DB
+	Orm *xorm.Engine
+)
+
+func init() {
+	Db = model.NewDB(connection)
+	Orm = Db.Engine
+
+	//...some settings for xorm
+	Orm.SetMapper(core.GonicMapper{})
+}

drivers/mysql.go

@@ -0,0 +1,34 @@
+package models
+
+import (
+	"github.com/go-xorm/builder"
+)
+
+type User struct {
+	Id         uint32
+	Cellphone  string
+	InsertTime uint32 `xorm:"created"`
+	ModifyTime uint32 `xorm:"updated"`
+}
+
+func CreateUser(cellphone string) *User {
+	user := new(User)
+	user.Cellphone = cellphone
+	_, err := Orm.Insert(user)
+	if err != nil {
+		return nil
+	}
+	return user
+}
+
+func GetUserById(id uint32) *User {
+	var user *User
+	Orm.Where("id=?", id).Limit(1, 0).Get(user)
+	return user
+}
+
+func GetUsersByIds(ids []uint32) *[]User {
+	var users *[]User
+	Orm.Where(builder.In("id", ids)).Find(users)
+	return users
+}