go zero gorm extension
- add the dependent
go get github.com/SpectatorNan/gorm-zero
- replace template/model in your project with gorm-zero/template/{goctl version}/model
- generate
goctl model mysql -src={patterns} -dir={dir} -cache --home ./template
import (
"github.com/SpectatorNan/gorm-zero/gormc/config/mysql"
)
type Config struct {
Mysql mysql.Mysql
...
}
import (
"github.com/SpectatorNan/gorm-zero/gormc/config/mysql"
)
func NewServiceContext(c config.Config) *ServiceContext {
db, err := mysql.Connect(c.Mysql)
if err != nil {
log.Fatal(err)
}
...
}
import (
"github.com/SpectatorNan/gorm-zero/gormc/config/pg"
)
type Config struct {
PgSql pg.PgSql
...
}
import (
"github.com/SpectatorNan/gorm-zero/gormc/config/pg"
)
func NewServiceContext(c config.Config) *ServiceContext {
db, err := pg.Connect(c.PgSql)
if err != nil {
log.Fatal(err)
}
...
}
gormzeroUsersIdKey := fmt.Sprintf("%s%v", cacheGormzeroUsersIdExpirePrefix, id)
var resp Users
err := m.QueryWithExpireCtx(ctx, &resp, gormzeroUsersIdKey, expire, func(conn *gorm.DB, v interface{}) error {
return conn.Model(&Users{}).Where("`id` = ?", id).First(&resp).Error
})
switch err {
case nil:
return &resp, nil
case gormc.ErrNotFound:
return nil, ErrNotFound
default:
return nil, err
}
gormzeroUsersIdKey := fmt.Sprintf("%s%v", cacheGormzeroUsersIdPrefix, id)
var resp Users
err := m.QueryCtx(ctx, &resp, gormzeroUsersIdKey, func(conn *gorm.DB, v interface{}) error {
return conn.Model(&Users{}).Where("`id` = ?", id).First(&resp).Error
})
switch err {
case nil:
return &resp, nil
case gormc.ErrNotFound:
return nil, ErrNotFound
default:
return nil, err
}
- go zero model example link: gorm-zero-example