ayo/gocommon
8
0
Fork 1
Code Issues Pull Requests Projects Releases Wiki Activity

helper 함수 추가

Browse Source
This commit is contained in:
mountain
2023-08-31 21:09:31 +09:00
parent 319dbf5a54
commit a011588509
1 changed files with 11 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
session/common.go
View File

@ -82,6 +82,11 @@ func publickey_to_storagekey(pk publickey) storagekey {
return storagekey(hex.EncodeToString(decoded[:])) return storagekey(hex.EncodeToString(decoded[:]))
} }
type Config struct {
SessionTTL int64 `json:"session_ttl"`
SessionStorage string `json:"session_storage"`
}
var errInvalidScheme = errors.New("storageAddr is not valid scheme") var errInvalidScheme = errors.New("storageAddr is not valid scheme")
func NewConsumer(ctx context.Context, storageAddr string, ttl time.Duration) (Consumer, error) { func NewConsumer(ctx context.Context, storageAddr string, ttl time.Duration) (Consumer, error) {
@ -96,6 +101,10 @@ func NewConsumer(ctx context.Context, storageAddr string, ttl time.Duration) (Co
return nil, errInvalidScheme return nil, errInvalidScheme
} }
func NewConsumerWithConfig(ctx context.Context, cfg Config) (Consumer, error) {
return NewConsumer(ctx, cfg.SessionStorage, time.Duration(cfg.SessionTTL)*time.Second)
}
func NewProvider(ctx context.Context, storageAddr string, ttl time.Duration) (Provider, error) { func NewProvider(ctx context.Context, storageAddr string, ttl time.Duration) (Provider, error) {
if strings.HasPrefix(storageAddr, "mongodb") { if strings.HasPrefix(storageAddr, "mongodb") {
return newProviderWithMongo(ctx, storageAddr, ttl) return newProviderWithMongo(ctx, storageAddr, ttl)
@ -108,7 +117,6 @@ func NewProvider(ctx context.Context, storageAddr string, ttl time.Duration) (Pr
return nil, errInvalidScheme return nil, errInvalidScheme
} }
type Config struct { func NewProviderWithConfig(ctx context.Context, cfg Config) (Provider, error) {
SessionTTL int64 `json:"session_ttl"` return NewProvider(ctx, cfg.SessionStorage, time.Duration(cfg.SessionTTL)*time.Second)
SessionStorage string `json:"session_storage"`
} }