golang框架中如何进行数据库读写分离

import (
    "database/sql"

    "github.com/jinzhu/gorm"
    _ "github.com/go-sql-driver/mysql"
)

func main() {
    masterDB, err := sql.Open("mysql", "user:password@tcp(master_db:3306)/db_name")
    if err != nil {
        panic(err)
    }
    slaveDB, err := sql.Open("mysql", "user:password@tcp(slave_db:3306)/db_name")
    if err != nil {
        panic(err)
    }
    db, err := gorm.Open("mysql", masterDB)
    if err != nil {
        panic(err)
    }
    db.SetMasterDB(masterDB)
    db.SetSlaveDB(slaveDB, SlaveDBModeReadOnly)
}

// 查询数据
type User struct {
    ID   int
    Name string
}
func findUser(db *gorm.DB) (*User, error) {
    user := &User{}
    if err := db.UseSlave().First(user, 1).Error; err != nil {
        return nil, err
    }
    return user, nil
}

// 更新数据
func updateUser(db *gorm.DB) error {
    user := &User{
        ID:   1,
        Name: "New Name",
    }
    if err := db.UseMaster().Save(user).Error; err != nil {
        return err
    }
    return nil
}