name.go 1.79 KB
Newer Older
1 2 3 4 5 6
package options

import (
	"time"
)

7 8 9 10
const (
	DefaultNameValidTime = 24 * time.Hour
)

11 12 13 14 15 16 17 18
type NamePublishSettings struct {
	ValidTime time.Duration
	Key       string
}

type NameResolveSettings struct {
	Recursive bool
	Local     bool
19
	Cache     bool
20 21 22 23 24 25 26
}

type NamePublishOption func(*NamePublishSettings) error
type NameResolveOption func(*NameResolveSettings) error

func NamePublishOptions(opts ...NamePublishOption) (*NamePublishSettings, error) {
	options := &NamePublishSettings{
27
		ValidTime: DefaultNameValidTime,
28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44
		Key:       "self",
	}

	for _, opt := range opts {
		err := opt(options)
		if err != nil {
			return nil, err
		}
	}

	return options, nil
}

func NameResolveOptions(opts ...NameResolveOption) (*NameResolveSettings, error) {
	options := &NameResolveSettings{
		Recursive: false,
		Local:     false,
45
		Cache:     true,
46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87
	}

	for _, opt := range opts {
		err := opt(options)
		if err != nil {
			return nil, err
		}
	}

	return options, nil
}

type NameOptions struct{}

func (api *NameOptions) WithValidTime(validTime time.Duration) NamePublishOption {
	return func(settings *NamePublishSettings) error {
		settings.ValidTime = validTime
		return nil
	}
}

func (api *NameOptions) WithKey(key string) NamePublishOption {
	return func(settings *NamePublishSettings) error {
		settings.Key = key
		return nil
	}
}

func (api *NameOptions) WithRecursive(recursive bool) NameResolveOption {
	return func(settings *NameResolveSettings) error {
		settings.Recursive = recursive
		return nil
	}
}

func (api *NameOptions) WithLocal(local bool) NameResolveOption {
	return func(settings *NameResolveSettings) error {
		settings.Local = local
		return nil
	}
}

88
func (api *NameOptions) WithCache(cache bool) NameResolveOption {
89
	return func(settings *NameResolveSettings) error {
90
		settings.Cache = cache
91 92 93
		return nil
	}
}