create.go

package redists

import (
	"context"
)

type cmdCreate struct {
	key             string
	retention       Duration
	encoding        *Encoding
	chunkSize       *int
	duplicatePolicy *DuplicatePolicy
	labels          map[string]string
}

func newCmdCreate(key string) *cmdCreate {
	return &cmdCreate{key: key}
}

func (c *cmdCreate) Name() string {
	return "TS.CREATE"
}

func (c *cmdCreate) Args() []interface{} {
	args := []interface{}{c.key}
	if c.retention != nil {
		args = append(args, optionNameRetention, c.retention.Milliseconds())
	}
	if c.encoding != nil {
		args = append(args, optionNameEncoding, string(*c.encoding))
	}
	if c.chunkSize != nil {
		args = append(args, optionNameChunkSize, *c.chunkSize)
	}
	if c.duplicatePolicy != nil {
		args = append(args, optionNameDuplicatePolicy, string(*c.duplicatePolicy))
	}
	if c.labels != nil {
		args = append(args, optionNameLabels)
		args = append(args, encodeLabels(c.labels)...)
	}
	return args
}

type OptionCreate func(cmd *cmdCreate)

// Create creates a new time-series.
func (c *Client) Create(ctx context.Context, key string, options ...OptionCreate) error {
	cmd := newCmdCreate(key)
	for i := range options {
		options[i](cmd)
	}
	_, err := c.d.Do(ctx, cmd.Name(), cmd.Args()...)
	return err
}

func CreateWithRetention(r Duration) OptionCreate {
	return func(cmd *cmdCreate) {
		cmd.retention = r
	}
}

func CreateWithEncoding(e Encoding) OptionCreate {
	return func(cmd *cmdCreate) {
		cmd.encoding = &e
	}
}

func CreateWithChunkSize(cs int) OptionCreate {
	return func(cmd *cmdCreate) {
		cmd.chunkSize = &cs
	}
}

func CreateWithDuplicatePolicy(dp DuplicatePolicy) OptionCreate {
	return func(cmd *cmdCreate) {
		cmd.duplicatePolicy = &dp
	}
}

func CreateWithLabels(ls Labels) OptionCreate {
	return func(cmd *cmdCreate) {
		cmd.labels = ls
	}
}