fix: db

apus-run · Aug 23, 2024 · ee279cf · ee279cf
1 parent c3bac4b
commit ee279cf
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/db/gorm/wrapper.go b/db/gorm/wrapper.go
@@ -29,6 +29,8 @@ type Transaction interface {
 	Execute(ctx context.Context, fn func(ctx context.Context) error) error
 }
 
+type DB = gorm.DB
+
 type Helper struct {
 	lock  *sync.RWMutex
 	group *singleflight.Group

diff --git a/db/redis/wrapper.go b/db/redis/wrapper.go
@@ -11,6 +11,10 @@ import (
 var _ Database = (*Helper)(nil)
 
 type (
+	// Cmdable is an alias of redis.Cmdable.
+	Cmdable = redis.Cmdable
+	// BoolCmd is an alias of redis.BoolCmd.
+	BoolCmd = redis.BoolCmd
 	// IntCmd is an alias of redis.IntCmd.
 	IntCmd = redis.IntCmd
 	// FloatCmd is an alias of redis.FloatCmd.