upgrade models

Author: XieBiao

examples/main.go

@@ -2,11 +2,11 @@ package main
 
 import (
 	"fmt"
-	"github.com/hhxsv5/go-db-proxy-api/models"
+	"github.com/hhxsv5/go-db-proxy-api/models/mdefault"
 )
 
 func main() {
-	user := models.CreateUser("18780207350")
+	user := mdefault.CreateUser("18780207350")
 	fmt.Println(user)
 	fmt.Printf("%p", user)
 }

models/base.go

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

models/mdb1/base.go

@@ -0,0 +1,21 @@
+package mdb1
+
+import (
+	_ "github.com/go-sql-driver/mysql" //remember: import xxx driver to init
+	"github.com/hhxsv5/go-db-proxy-api/models"
+	"github.com/hhxsv5/go-db-proxy-api/core"
+	"github.com/go-xorm/xorm"
+)
+
+//db1 connection in file db.toml
+const conn = "db1"
+
+var (
+	db  *core.DB
+	orm *xorm.Engine
+)
+
+func init() {
+	db = models.NewDB(conn)
+	orm = db.Engine
+}

models/user.go renamed to models/mdb1/user.go

@@ -1,4 +1,4 @@
-package models
+package mdb1
 
 import (
 	"github.com/go-xorm/builder"
@@ -14,7 +14,7 @@ type User struct {
 func CreateUser(cellphone string) *User {
 	user := new(User)
 	user.Cellphone = cellphone
-	_, err := Orm.Insert(user)
+	_, err := orm.Insert(user)
 	if err != nil {
 		return nil
 	}
@@ -23,12 +23,12 @@ func CreateUser(cellphone string) *User {
 
 func GetUserById(id uint32) *User {
 	var user *User
-	Orm.Where("id=?", id).Limit(1, 0).Get(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)
+	orm.Where(builder.In("id", ids)).Find(users)
 	return users
 }

models/mdb2/base.go

@@ -0,0 +1,21 @@
+package mdb2
+
+import (
+	_ "github.com/go-sql-driver/mysql" //remember: import xxx driver to init
+	"github.com/hhxsv5/go-db-proxy-api/models"
+	"github.com/hhxsv5/go-db-proxy-api/core"
+	"github.com/go-xorm/xorm"
+)
+
+//db2 connection in file db.toml
+const conn = "db2"
+
+var (
+	db  *core.DB
+	orm *xorm.Engine
+)
+
+func init() {
+	db = models.NewDB(conn)
+	orm = db.Engine
+}

models/mdb2/user.go

@@ -0,0 +1,34 @@
+package mdb2
+
+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
+}

models/mdefault/base.go

@@ -0,0 +1,21 @@
+package mdefault
+
+import (
+	_ "github.com/go-sql-driver/mysql" //remember: import xxx driver to init
+	"github.com/hhxsv5/go-db-proxy-api/models"
+	"github.com/hhxsv5/go-db-proxy-api/core"
+	"github.com/go-xorm/xorm"
+)
+
+//default connection in file db.toml
+const conn = "default"
+
+var (
+	db  *core.DB
+	orm *xorm.Engine
+)
+
+func init() {
+	db = models.NewDB(conn)
+	orm = db.Engine
+}

models/mdefault/user.go

@@ -0,0 +1,34 @@
+package mdefault
+
+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
+}