zishuo/eagle
1
0
Fork 0
mirror of https://github.com/go-eagle/eagle.git synced 2025-09-26 20:41:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

chore: optimzie_repo_template (#163)

Browse Source
This commit is contained in:
Richard
2024-12-06 09:01:05 +08:00
committed by GitHub
parent 7ed3b22d20
commit 09179bcdac
2 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
cmd/eagle/internal/repo/add/template.go
View File

@@ -45,14 +45,14 @@ type {{.LcName}}Repo struct {
tracer trace.Tracer
{{- if eq .WithCache true }}
cache cache.{{.Name}}Cache
{{- end }}
localCache localCache.Cache
sg singleflight.Group
{{- end }}
}
{{- if .WithCache }}
// New{{.Name}} new a repository and return
func New{{.Name}}(db *dal.DBClient, cache cache.{{.Name}}Cache) {{.Name}}Repo {
// New{{.Name}}Repo new a repository and return
func New{{.Name}}Repo(db *dal.DBClient, cache cache.{{.Name}}Cache) {{.Name}}Repo {
return &{{.LcName}}Repo{
db: db,
tracer: otel.Tracer("{{.LcName}}"),
@@ -62,8 +62,8 @@ func New{{.Name}}(db *dal.DBClient, cache cache.{{.Name}}Cache) {{.Name}}Repo {
}
}
{{- else }}
// New{{.Name}} new a repository and return
func New{{.Name}}(db *dal.DBClient) {{.Name}}Repo {
// New{{.Name}}Repo new a repository and return
func New{{.Name}}Repo(db *dal.DBClient) {{.Name}}Repo {
return &{{.LcName}}Repo{
db: db,
tracer: otel.Tracer("{{.LcName}}Repo"),
@@ -135,7 +135,7 @@ func (r *{{.LcName}}Repo) Get{{.Name}}(ctx context.Context, id int64) (ret *mode
// 避免缓存击穿(瞬间有大量请求过来)
val, err, _ := r.sg.Do("sg:{{.LcName}}:"+cast.ToString(id), func() (interface{}, error) {
// read db or rpc
data, err := dao.{{.Name}}Model.WithContext(ctx).Where(dao.{{.Name}}Model.ID.Eq(id)).First()
data, err := dao.{{.Name}}Model.WithContext(ctx).Where(dao.{{.Name}}Model.ID.Eq(id)).Last()
if err != nil {
// cache not found and set empty cache to avoid 缓存穿透
// Note: 如果缓存空数据太多，会大大降低缓存命中率，可以改为使用布隆过滤器
@@ -168,7 +168,7 @@ func (r *{{.LcName}}Repo) Get{{.Name}}(ctx context.Context, id int64) (ret *mode
return val.(*model.{{.Name}}Model), nil
{{- else }}
// read db
ret, err = dao.{{.Name}}Model.WithContext(ctx).Where(dao.{{.Name}}Model.ID.In(ids...)).Find()
ret, err = dao.{{.Name}}Model.WithContext(ctx).Where(dao.{{.Name}}Model.ID.Eq(id)).Last()
if err != nil {
return nil, err
}
@@ -212,8 +212,8 @@ func (r *{{.LcName}}Repo) BatchGet{{.Name}}(ctx context.Context, ids []int64) (r
return ret, nil
{{- else }}
// read db
items := make([]*model.{{.Name}}Model, 0)
items, err := dao.{{.Name}}Model.WithContext(ctx).Where(dao.{{.Name}}Model.ID.In(ids...)).Find()
items := make([]*model.{{.Name}}Model, len(ids))
items, err = dao.{{.Name}}Model.WithContext(ctx).Where(dao.{{.Name}}Model.ID.In(ids...)).Find()
if err != nil {
return
}

2
cmd/eagle/main.go
View File

@@ -19,7 +19,7 @@ import (
var (
// Version is the version of the compiled software.
Version = "v1.0.1"
Version = "v1.0.2"
rootCmd = &cobra.Command{
Use: "eagle",