digimon/library-go
digimon
/
library-go
3
0
Fork 0
Code Issues Pull Requests 1 Packages Projects Releases 27 Wiki Activity

feat: fix doc-db-port

This commit is contained in:
王性驊 2024-12-26 14:25:45 +08:00
parent 24e302990a
commit 240027443a
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mongo/doc-db-with-cache.go
View File

@ -15,8 +15,8 @@ type DocumentDBWithCache struct {
Cache cache.Cache Cache cache.Cache
} }
func MustDocumentDBWithCache(conf *Conf, cacheConf cache.CacheConf, dbOpts []mon.Option, cacheOpts []cache.Option) (DocumentDBWithCacheUseCase, error) { func MustDocumentDBWithCache(conf *Conf, collection string, cacheConf cache.CacheConf, dbOpts []mon.Option, cacheOpts []cache.Option) (DocumentDBWithCacheUseCase, error) {
documentDB, err := NewDocumentDB(conf, dbOpts...) documentDB, err := NewDocumentDB(conf, collection, dbOpts...)
if err != nil { if err != nil {
return nil, fmt.Errorf("failed to initialize DocumentDB: %w", err) return nil, fmt.Errorf("failed to initialize DocumentDB: %w", err)
} }

4
mongo/doc-db.go
View File

@ -20,7 +20,7 @@ type DocumentDB struct {
Mon *mon.Model Mon *mon.Model
} }
func NewDocumentDB(config *Conf, opts ...mon.Option) (DocumentDBUseCase, error) { func NewDocumentDB(config *Conf, collection string, opts ...mon.Option) (DocumentDBUseCase, error) {
authenticationURI := "" authenticationURI := ""
if config.User != "" { if config.User != "" {
authenticationURI = fmt.Sprintf( authenticationURI = fmt.Sprintf(
@ -50,7 +50,7 @@ func NewDocumentDB(config *Conf, opts ...mon.Option) (DocumentDBUseCase, error)
opts = append(opts, intOpt) opts = append(opts, intOpt)
logx.Infof("[DocumentDB] Try to connect document db `%s`", printConnectUri) logx.Infof("[DocumentDB] Try to connect document db `%s`", printConnectUri)
client, err := mon.NewModel(connectionURI, config.Database, config.Database, opts...) client, err := mon.NewModel(connectionURI, config.Database, collection, opts...)
if err != nil { if err != nil {
return nil, err return nil, err
} }