github_mirrors/go-zero
1
0
Fork 0
mirror of https://github.com/zeromicro/go-zero.git synced 2025-02-03 00:38:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

Feature: support adding custom cache to mongoc and sqlc (#1313)

Browse Source 
* merge

* Feature: support adding custom cache to mongoc and sqlc
This commit is contained in:
MarkJoyMa 2021-12-18 22:45:07 +08:00 committed by GitHub
parent b299f350be
commit 3e6c217408
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 22 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
core/stores/mongoc/cachedmodel.go
View File

@ -52,6 +52,16 @@ func NewModel(url, collection string, conf cache.CacheConf, opts ...cache.Option
})
}
// NewModelWithCache returns a Model with a custom cache.
func NewModelWithCache(url, collection string, c cache.Cache) (*Model, error) {
if c == nil {
log.Fatal("Invalid cache component")
}
return createModel(url, collection, c, func(collection mongo.Collection) CachedCollection {
return newCollection(collection, c)
})
}
// Count returns the count of given query.
func (mm *Model) Count(query interface{}) (int, error) {
return mm.executeInt(func(c CachedCollection) (int, error) {

12
core/stores/sqlc/cachedsql.go
View File

@ -2,6 +2,7 @@ package sqlc
import (
"database/sql"
"log"
"time"
"github.com/tal-tech/go-zero/core/stores/cache"
@ -55,6 +56,17 @@ func NewConn(db sqlx.SqlConn, c cache.CacheConf, opts ...cache.Option) CachedCon
}
}
// NewConnWithCache returns a CachedConn with a custom cache.
func NewConnWithCache(db sqlx.SqlConn, c cache.Cache) CachedConn {
if c == nil {
log.Fatal("Invalid cache component")
}
return CachedConn{
db: db,
cache: c,
}
}
// DelCache deletes cache with keys.
func (cc CachedConn) DelCache(keys ...string) error {
return cc.cache.Del(keys...)