add: go generate

basic-layout/.gitignore

@@ -19,4 +19,6 @@
 vendor/
 vendor
 /Godeps/
-/scripts/docker/mysql/data
+/scripts/docker/mysql/data
+go.work
+go.work.sum

basic-layout/Makefile

@@ -24,16 +24,10 @@ check:
 # 生成 mock 文件
 .PHONY: mock
 mock:
-	@mockgen -source=./internal/service/user.go -package=svcmocks -destination=./internal/service/mocks/user.mock.go
-	@mockgen -source=./internal/repository/user.go -package=repomocks -destination=./internal/repository/mocks/user.mock.go
-	@mockgen -source=./internal/repository/dao/user.go -package=daomocks -destination=./internal/repository/dao/mocks/user.mock.go
-	@mockgen -source=./internal/repo/cache/code.go -package=cachemocks -destination=./internal/repo/cache/mocks/code.mock.go
-	@mockgen -package=redismocks -destination=./internal/repo/cache/redismocks/cmd.mock.go github.com/redis/go-redis/v9 Cmdable
-
+	@go generate -tags=wireinject ./...
 	@$(MAKE) tidy
 	@echo "go mock finished"
 
-
 # 单元测试
 .PHONY: test
 	@go test -race ./..

basic-layout/go.mod

@@ -16,7 +16,9 @@ require (
 	github.com/redis/go-redis/v9 v9.3.0
 	github.com/spf13/pflag v1.0.5
 	github.com/spf13/viper v1.17.0
+	github.com/stretchr/testify v1.8.4
 	go.uber.org/automaxprocs v1.5.3
+	go.uber.org/mock v0.3.0
 	gopkg.in/natefinch/lumberjack.v2 v2.2.1
 	gorm.io/driver/mysql v1.5.2
 )
@@ -67,7 +69,6 @@ require (
 	github.com/sourcegraph/conc v0.3.0 // indirect
 	github.com/spf13/afero v1.10.0 // indirect
 	github.com/spf13/cast v1.5.1 // indirect
-	github.com/stretchr/testify v1.8.4 // indirect
 	github.com/subosito/gotenv v1.6.0 // indirect
 	github.com/twitchyliquid64/golang-asm v0.15.1 // indirect
 	github.com/ugorji/go/codec v1.2.11 // indirect

basic-layout/go.sum

@@ -313,6 +313,8 @@ go.uber.org/atomic v1.11.0/go.mod h1:LUxbIzbOniOlMKjJjyPfpl4v+PKK2cNJn91OQbhoJI0
 go.uber.org/automaxprocs v1.5.3 h1:kWazyxZUrS3Gs4qUpbwo5kEIMGe/DAvi5Z4tl2NW4j8=
 go.uber.org/automaxprocs v1.5.3/go.mod h1:eRbA25aqJrxAbsLO0xy5jVwPt7FQnRgjW+efnwa1WM0=
 go.uber.org/goleak v1.2.0 h1:xqgm/S+aQvhWFTtR0XK3Jvg7z8kGV8P4X14IzwN3Eqk=
+go.uber.org/mock v0.3.0 h1:3mUxI1No2/60yUYax92Pt8eNOEecx2D3lcXZh2NEZJo=
+go.uber.org/mock v0.3.0/go.mod h1:a6FSlNadKUHUa9IP5Vyt1zh4fC7uAwxMutEAscFbkZc=
 go.uber.org/multierr v1.10.0 h1:S0h4aNzvfcFsC3dRF1jLoaov7oRaKqRGC/pUEJ2yvPQ=
 go.uber.org/multierr v1.10.0/go.mod h1:20+QtiLqy0Nd6FdQB9TLXag12DsQkrbs3htMFfDN80Y=
 go.uber.org/zap v1.26.0 h1:sI7k6L95XOKS281NhVKOFCUNIvv9e0w4BF8N3u+tCRo=

basic-layout/internal/core/repo/cache/user.go

@@ -1 +1,60 @@
 package cache
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"time"
+
+	"github.com/redis/go-redis/v9"
+
+	"basic-layout/internal/core/domain/entity"
+)
+
+//go:generate mockgen -source=./user.go -package=cachemocks -destination=mocks/user.mock.go UserCache
+type UserCache interface {
+	Delete(ctx context.Context, id uint64) error
+	Get(ctx context.Context, id uint64) (entity.User, error)
+	Set(ctx context.Context, u entity.User) error
+}
+
+type RedisUserCache struct {
+	cmd redis.Cmdable
+	// 过期时间
+	expiration time.Duration
+}
+
+func NewRedisUserCache(cmd redis.Cmdable) UserCache {
+	return &RedisUserCache{
+		cmd:        cmd,
+		expiration: time.Minute * 15,
+	}
+}
+func (cache *RedisUserCache) Delete(ctx context.Context, id uint64) error {
+	return cache.cmd.Del(ctx, cache.key(id)).Err()
+}
+
+func (cache *RedisUserCache) Get(ctx context.Context, id uint64) (entity.User, error) {
+	key := cache.key(id)
+	data, err := cache.cmd.Get(ctx, key).Result()
+	if err != nil {
+		return entity.User{}, err
+	}
+	// 反序列化回来
+	var u entity.User
+	err = json.Unmarshal([]byte(data), &u)
+	return u, err
+}
+
+func (cache *RedisUserCache) Set(ctx context.Context, u entity.User) error {
+	data, err := json.Marshal(u)
+	if err != nil {
+		return err
+	}
+	key := cache.key(u.ID)
+	return cache.cmd.Set(ctx, key, data, cache.expiration).Err()
+}
+
+func (cache *RedisUserCache) key(id uint64) string {
+	return fmt.Sprintf("user:info:%d", id)
+}

basic-layout/internal/core/repo/dao/user.go

@@ -12,6 +12,8 @@ import (
 var ErrDataNotFound = gorm.ErrRecordNotFound
 
 // UserDAO ... 数据访问层相关接口定义, 使用 DB 风格的命名
+//
+//go:generate mockgen -source=./user.go -package=daomocks -destination=mocks/user.mock.go UserDAO
 type UserDAO interface {
 	FindByID(ctx context.Context, id uint64) (*model.User, error)
 }

basic-layout/internal/core/repo/user.go

@@ -8,6 +8,7 @@ import (
 	"basic-layout/pkg/logger"
 )
 
+//go:generate mockgen -source=./user.go -package=repomocks -destination=mocks/user.mock.go UserRepository
 type UserRepository interface {
 	FindByID(ctx context.Context, id uint64) (*entity.User, error)
 }

basic-layout/internal/core/svc/user.go

@@ -8,6 +8,7 @@ import (
 	"basic-layout/pkg/logger"
 )
 
+//go:generate mockgen -source=./user.go -package=svcmocks -destination=mocks/user.mock.go UserService
 type UserService interface {
 	GetByID(ctx context.Context, id uint64) (*entity.User, error)
 }

gaea-layout/.gitignore

@@ -19,4 +19,6 @@
 vendor/
 vendor
 /Godeps/
-/scripts/docker/mysql/data
+/scripts/docker/mysql/data
+go.work
+go.work.sum

gaia-layout/.gitignore

@@ -19,4 +19,6 @@
 vendor/
 vendor
 /Godeps/
-/scripts/docker/mysql/data
+/scripts/docker/mysql/data
+go.work
+go.work.sum