object.go 1.76 KB
Newer Older
Łukasz Magiera's avatar
Łukasz Magiera committed
1 2 3 4 5 6
package options

type ObjectNewSettings struct {
	Type string
}

7 8
type ObjectPutSettings struct {
	InputEnc string
Łukasz Magiera's avatar
Łukasz Magiera committed
9
	DataType string
10 11
}

Łukasz Magiera's avatar
Łukasz Magiera committed
12 13 14 15 16
type ObjectAddLinkSettings struct {
	Create bool
}

type ObjectNewOption func(*ObjectNewSettings) error
17
type ObjectPutOption func(*ObjectPutSettings) error
Łukasz Magiera's avatar
Łukasz Magiera committed
18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33
type ObjectAddLinkOption func(*ObjectAddLinkSettings) error

func ObjectNewOptions(opts ...ObjectNewOption) (*ObjectNewSettings, error) {
	options := &ObjectNewSettings{
		Type: "empty",
	}

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

34 35 36
func ObjectPutOptions(opts ...ObjectPutOption) (*ObjectPutSettings, error) {
	options := &ObjectPutSettings{
		InputEnc: "json",
Łukasz Magiera's avatar
Łukasz Magiera committed
37
		DataType: "text",
38 39 40 41 42 43 44 45 46 47 48
	}

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

Łukasz Magiera's avatar
Łukasz Magiera committed
49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71
func ObjectAddLinkOptions(opts ...ObjectAddLinkOption) (*ObjectAddLinkSettings, error) {
	options := &ObjectAddLinkSettings{
		Create: false,
	}

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

type ObjectOptions struct{}

func (api *ObjectOptions) WithType(t string) ObjectNewOption {
	return func(settings *ObjectNewSettings) error {
		settings.Type = t
		return nil
	}
}

72 73 74 75 76 77 78
func (api *ObjectOptions) WithInputEnc(e string) ObjectPutOption {
	return func(settings *ObjectPutSettings) error {
		settings.InputEnc = e
		return nil
	}
}

Łukasz Magiera's avatar
Łukasz Magiera committed
79 80 81 82 83 84 85
func (api *ObjectOptions) WithDataType(t string) ObjectPutOption {
	return func(settings *ObjectPutSettings) error {
		settings.DataType = t
		return nil
	}
}

Łukasz Magiera's avatar
Łukasz Magiera committed
86 87 88 89 90 91
func (api *ObjectOptions) WithCreate(create bool) ObjectAddLinkOption {
	return func(settings *ObjectAddLinkSettings) error {
		settings.Create = create
		return nil
	}
}