change store options
This commit is contained in:
@@ -1,23 +1,60 @@
|
||||
package cloudflare
|
||||
|
||||
import (
|
||||
"github.com/micro/go-micro/config/options"
|
||||
"context"
|
||||
|
||||
"github.com/micro/go-micro/store"
|
||||
)
|
||||
|
||||
func getOption(ctx context.Context, key string) string {
|
||||
if ctx == nil {
|
||||
return ""
|
||||
}
|
||||
val, ok := ctx.Value(key).(string)
|
||||
if !ok {
|
||||
return ""
|
||||
}
|
||||
return val
|
||||
}
|
||||
|
||||
func getToken(ctx context.Context) string {
|
||||
return getOption(ctx, "CF_API_TOKEN")
|
||||
}
|
||||
|
||||
func getAccount(ctx context.Context) string {
|
||||
return getOption(ctx, "CF_ACCOUNT_ID")
|
||||
}
|
||||
|
||||
func getNamespace(ctx context.Context) string {
|
||||
return getOption(ctx, "KV_NAMESPACE_ID")
|
||||
}
|
||||
|
||||
// Token sets the cloudflare api token
|
||||
func Token(t string) options.Option {
|
||||
// TODO: change to store.cf.api_token
|
||||
return options.WithValue("CF_API_TOKEN", t)
|
||||
func Token(t string) store.Option {
|
||||
return func(o *store.Options) {
|
||||
if o.Context == nil {
|
||||
o.Context = context.Background()
|
||||
}
|
||||
o.Context = context.WithValue(o.Context, "CF_API_TOKEN", t)
|
||||
}
|
||||
}
|
||||
|
||||
// Account sets the cloudflare account id
|
||||
func Account(id string) options.Option {
|
||||
// TODO: change to store.cf.account_id
|
||||
return options.WithValue("CF_ACCOUNT_ID", id)
|
||||
func Account(id string) store.Option {
|
||||
return func(o *store.Options) {
|
||||
if o.Context == nil {
|
||||
o.Context = context.Background()
|
||||
}
|
||||
o.Context = context.WithValue(o.Context, "CF_ACCOUNT_ID", id)
|
||||
}
|
||||
}
|
||||
|
||||
// Namespace sets the KV namespace
|
||||
func Namespace(ns string) options.Option {
|
||||
// TODO: change to store.cf.namespace
|
||||
return options.WithValue("KV_NAMESPACE_ID", ns)
|
||||
func Namespace(ns string) store.Option {
|
||||
return func(o *store.Options) {
|
||||
if o.Context == nil {
|
||||
o.Context = context.Background()
|
||||
}
|
||||
o.Context = context.WithValue(o.Context, "KV_NAMESPACE_ID", ns)
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user