golang中repository的使用

可以添加一个repository来访问数据可以通过以下方法来做:
1,先在/Users/code/gopath/src/study/gorm/db.go添加以下代码:

package mysqltools
import (
    "fmt"
    "github.com/jinzhu/gorm"
    _ "github.com/jinzhu/gorm/dialects/mysql"
    "log"
    "sync"
)
/*
* MysqlConnectionPool
* 数据库连接操作库
* 基于gorm封装开发
 */
type MysqlConnectionPool struct {
}
var instance *MysqlConnectionPool
var once sync.Once
var db *gorm.DB
var errDB error
func GetInstance() *MysqlConnectionPool {
    once.Do(func() {
        instance = &MysqlConnectionPool{}
    })
    return instance
}
/*
* @fuc 初始化数据库连接(可在mail()适当位置调用)
*/
func (m *MysqlConnectionPool) InitDataPool() (isSuccess bool) {
    db, errDB = gorm.Open(
        "mysql",
        "root:root@tcp(localhost:3306)/default?charset=utf8&parseTime=True&loc=Local")
    fmt.Println(errDB)
    if errDB != nil {
        log.Fatal(errDB)
        return false
    }
    //关闭数据库，db会被多个goroutine共享，可以不调用
    // defer db.Close()
    return true
}
/*
* @fuc  对外获取数据库连接对象db
*/
func (m *MysqlConnectionPool) GetMysqlDB() (dbCon *gorm.DB) {
    return db
}

2,在/Users/code/gopath/src/study/model/student.go添加repository(现在和model一起)

package studentModel
import (
    "errors"
    "study/gorm"
)
type Student struct {
    //gorm.Model
    Id        int `gorm:"primary_key;auto_increment"`
    Telephone string
    FirstName string
    LastName  string
}
type Interface interface {
    SelectById(id int64) (stu Student, err error)
}
type StudentRepository struct {
    Interface
}
func (r *StudentRepository) SelectById(id int64) (stu []Student, err error) {
    // select
    var userRet []Student
    db := mysqltools.GetInstance().GetMysqlDB()
    // 使用id获取记录
    e := db.Where("id = ?", id).First(&userRet).Error
    if e != nil {
        return userRet, errors.New("查询失败")
    }
    return userRet, nil
}

3,然后在/Users/code/gopath/src/study/test/abc.go里添加如下代码:

package main
import (
    "fmt"
    "log"
    "os"
    "study/gorm"
    "study/model"
)
func init() {
    // init database pool
    isSuccess := mysqltools.GetInstance().InitDataPool()
    if !isSuccess {
        log.Println("init database pool failure...")
        os.Exit(1)
    }
}
func main() {
    stu := studentModel.StudentRepository{}
    data, err := stu.SelectById(1)
    if err != nil {
        fmt.Println(err);
    }
    fmt.Println(data)
}

4,这样子就可以了,在/Users/code/gopath/src/study/test跑下go run abc.go,会看到以下结果:

<nil>
[{1 11 LejuRobot test}]

一些说明:
gopath目录: GOPATH="/Users/code/gopath",这个大家对应着看.

golang中repository的使用

微信号：腾讯云折扣返现

技术讨论QQ群： 863102326

作者: catherine

搜索

标签