viney-shih/go-cache

package cache

// MarshalFunc specifies the algorithm during marshaling the value to bytes.
// The default is json.Marshal.
type MarshalFunc func(interface{}) ([]byte, error)

// UnmarshalFunc specifies the algorithm during unmarshaling the bytes to the value.
// The default is json.Unmarshal
type UnmarshalFunc func([]byte, interface{}) error

// FactoryOptions is an alias for functional argument.
type FactoryOptions func(opts *factoryOptions)

// factoryOptions contains all options which will be applied when calling NewFactory().
type factoryOptions struct {
    marshalFunc   MarshalFunc
    unmarshalFunc UnmarshalFunc
    onCacheHit    func(prefix string, key string, count int)
    onCacheMiss   func(prefix string, key string, count int)
    onLCCostAdd   func(prefix string, key string, cost int)
    onLCCostEvict func(prefix string, key string, cost int)
    pubsub        Pubsub
}

// WithMarshalFunc sets up the specified marshal function.
// Needs to consider with unmarshal function at the same time.
func WithMarshalFunc(f MarshalFunc) FactoryOptions {
    return func(opts *factoryOptions) {
        opts.marshalFunc = f
    }
}

// WithUnmarshalFunc sets up the specified unmarshal function.
// Needs to consider with marshal function at the same time.
func WithUnmarshalFunc(f UnmarshalFunc) FactoryOptions {
    return func(opts *factoryOptions) {
        opts.unmarshalFunc = f
    }
}

// WithPubSub is used to evict keys in local cache
func WithPubSub(pb Pubsub) FactoryOptions {
    return func(opts *factoryOptions) {
        opts.pubsub = pb
    }
}

// OnCacheHitFunc sets up the callback function on cache hitted
func OnCacheHitFunc(f func(prefix string, key string, count int)) FactoryOptions {
    return func(opts *factoryOptions) {
        opts.onCacheHit = f
    }
}

// OnCacheMissFunc sets up the callback function on cache missed
func OnCacheMissFunc(f func(prefix string, key string, count int)) FactoryOptions {
    return func(opts *factoryOptions) {
        opts.onCacheMiss = f
    }
}

// OnLocalCacheCostAddFunc sets up the callback function on adding the cost of key in local cache
func OnLocalCacheCostAddFunc(f func(prefix string, key string, cost int)) FactoryOptions {
    return func(opts *factoryOptions) {
        opts.onLCCostAdd = f
    }
}

// OnLocalCacheCostEvictFunc sets up the callback function on evicting the cost of key in local cache
func OnLocalCacheCostEvictFunc(f func(prefix string, key string, cost int)) FactoryOptions {
    return func(opts *factoryOptions) {
        opts.onLCCostEvict = f
    }
}

func loadFactoryOptions(options ...FactoryOptions) *factoryOptions {
    opts := &factoryOptions{}
    for _, option := range options {
        option(opts)
    }

    return opts
}