mirror of
https://github.com/openimsdk/open-im-server.git
synced 2025-04-06 04:15:46 +08:00
59 lines
2.0 KiB
Go
59 lines
2.0 KiB
Go
package controller
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/openimsdk/open-im-server/v3/pkg/common/storage/cache"
|
|
"github.com/openimsdk/open-im-server/v3/pkg/common/storage/database"
|
|
"github.com/openimsdk/open-im-server/v3/pkg/common/storage/model"
|
|
"github.com/openimsdk/tools/db/pagination"
|
|
"github.com/openimsdk/tools/db/tx"
|
|
)
|
|
|
|
type ClientConfigDatabase interface {
|
|
SetUserConfig(ctx context.Context, userID string, config map[string]string) error
|
|
GetUserConfig(ctx context.Context, userID string) (map[string]string, error)
|
|
DelUserConfig(ctx context.Context, userID string, keys []string) error
|
|
GetUserConfigPage(ctx context.Context, userID string, key string, pagination pagination.Pagination) (int64, []*model.ClientConfig, error)
|
|
}
|
|
|
|
func NewClientConfigDatabase(db database.ClientConfig, cache cache.ClientConfigCache, tx tx.Tx) ClientConfigDatabase {
|
|
return &clientConfigDatabase{
|
|
tx: tx,
|
|
db: db,
|
|
cache: cache,
|
|
}
|
|
}
|
|
|
|
type clientConfigDatabase struct {
|
|
tx tx.Tx
|
|
db database.ClientConfig
|
|
cache cache.ClientConfigCache
|
|
}
|
|
|
|
func (x *clientConfigDatabase) SetUserConfig(ctx context.Context, userID string, config map[string]string) error {
|
|
return x.tx.Transaction(ctx, func(ctx context.Context) error {
|
|
if err := x.db.Set(ctx, userID, config); err != nil {
|
|
return err
|
|
}
|
|
return x.cache.DeleteUserCache(ctx, []string{userID})
|
|
})
|
|
}
|
|
|
|
func (x *clientConfigDatabase) GetUserConfig(ctx context.Context, userID string) (map[string]string, error) {
|
|
return x.cache.GetUserConfig(ctx, userID)
|
|
}
|
|
|
|
func (x *clientConfigDatabase) DelUserConfig(ctx context.Context, userID string, keys []string) error {
|
|
return x.tx.Transaction(ctx, func(ctx context.Context) error {
|
|
if err := x.db.Del(ctx, userID, keys); err != nil {
|
|
return err
|
|
}
|
|
return x.cache.DeleteUserCache(ctx, []string{userID})
|
|
})
|
|
}
|
|
|
|
func (x *clientConfigDatabase) GetUserConfigPage(ctx context.Context, userID string, key string, pagination pagination.Pagination) (int64, []*model.ClientConfig, error) {
|
|
return x.db.GetPage(ctx, userID, key, pagination)
|
|
}
|