feat: Optimize curd

fynntang · Aug 29, 2023 · 59aa60a · 59aa60a
1 parent c0ae0d5
commit 59aa60a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/protoc-gen-go-crud/repositories/template.go b/protoc-gen-go-crud/repositories/template.go
@@ -28,11 +28,11 @@ func New{{.ServiceType}}Repo() I{{.ServiceType}}Repo{
 	return &{{.ServiceType}}Repo{}
 }
 
-func ({{$firstLetter}} *{{.ServiceType}}Repo) log(ctx context.Context) *zap.SugaredLogger {
+func (r *{{.ServiceType}}Repo) log(ctx context.Context) *zap.SugaredLogger {
 	return global.Logger(ctx).Named("{{.ServiceType}}Repo")
 }
 
-func ({{$firstLetter}} *{{.ServiceType}}Repo) DB(ctx context.Context) *gorm.DB {
+func (r *{{.ServiceType}}Repo) DB(ctx context.Context) *gorm.DB {
 	return global.DBFromContext(ctx).Model(&entities.{{.ServiceType}}{})
 }