misc_commands.go

// Copyright 2015-present Basho Technologies, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package riak

import (
	"fmt"
	"reflect"

	rpbRiak "github.com/basho/riak-go-client/rpb/riak"
	proto "github.com/golang/protobuf/proto"
)

// PingCommandBuilder is the command builder required for PingCommand
type PingCommandBuilder struct {
}

// Build validates the configuration options provided then builds the command
func (builder *PingCommandBuilder) Build() (Command, error) {
	return &PingCommand{}, nil
}

// PingCommand is used to verify Riak is online and reachable
type PingCommand struct {
	commandImpl
	retryableCommandImpl
}

// Name identifies this command
func (cmd *PingCommand) Name() string {
	return cmd.getName("Ping")
}

func (cmd *PingCommand) getRequestCode() byte {
	return rpbCode_RpbPingReq
}

func (cmd *PingCommand) constructPbRequest() (msg proto.Message, err error) {
	return nil, nil
}

func (cmd *PingCommand) onSuccess(msg proto.Message) error {
	cmd.success = true
	return nil
}

func (cmd *PingCommand) getResponseCode() byte {
	return rpbCode_RpbPingResp
}

func (cmd *PingCommand) getResponseProtobufMessage() proto.Message {
	return nil
}

// GetServerInfoResponse contains the response data for Riak server information
type GetServerInfoResponse struct {
	Node          string
	ServerVersion string
}

// GetServerInfoCommand is used to get Riak server information
type GetServerInfoCommand struct {
	commandImpl
	Response *GetServerInfoResponse
}

// Name identifies this command
func (cmd *GetServerInfoCommand) Name() string {
	return cmd.getName("GetServerInfo")
}

func (cmd *GetServerInfoCommand) getRequestCode() byte {
	return rpbCode_RpbGetServerInfoReq
}

func (cmd *GetServerInfoCommand) constructPbRequest() (msg proto.Message, err error) {
	return nil, nil
}

func (cmd *GetServerInfoCommand) onSuccess(msg proto.Message) error {
	cmd.success = true
	if msg == nil {
		cmd.success = false
	} else {
		if rpbResp, ok := msg.(*rpbRiak.RpbGetServerInfoResp); ok {
			cmd.Response = &GetServerInfoResponse{
				Node:          string(rpbResp.GetNode()),
				ServerVersion: string(rpbResp.GetServerVersion()),
			}
		} else {
			return fmt.Errorf("[GetServerInfoCommand] could not convert %v to RpbGetServerInfoResp", reflect.TypeOf(msg))
		}
	}
	return nil
}

func (cmd *GetServerInfoCommand) getResponseCode() byte {
	return rpbCode_RpbGetServerInfoResp
}

func (cmd *GetServerInfoCommand) getResponseProtobufMessage() proto.Message {
	return &rpbRiak.RpbGetServerInfoResp{}
}

type startTlsCommand struct {
	commandImpl
}

// Name identifies this command
func (cmd *startTlsCommand) Name() string {
	return cmd.getName("StartTls")
}

func (cmd *startTlsCommand) constructPbRequest() (msg proto.Message, err error) {
	return nil, nil
}

func (cmd *startTlsCommand) onSuccess(msg proto.Message) error {
	cmd.success = true
	return nil
}

func (cmd *startTlsCommand) getRequestCode() byte {
	return rpbCode_RpbStartTls
}

func (cmd *startTlsCommand) getResponseCode() byte {
	return rpbCode_RpbStartTls
}

func (cmd *startTlsCommand) getResponseProtobufMessage() proto.Message {
	return nil
}

// Types for bucket type and bucket properties

// ReplMode contains the replication mode
type ReplMode int32

// Convenience constants for maintaining the replication mode
const (
	FALSE    ReplMode = 0
	REALTIME ReplMode = 1
	FULLSYNC ReplMode = 2
	TRUE     ReplMode = 3
)

// CommitHook object is used when fetching or updating pre- or post- commit hook bucket properties
// on Riak
type CommitHook struct {
	Name   string
	ModFun *ModFun
}

func getHooksFrom(rpbHooks []*rpbRiak.RpbCommitHook) []*CommitHook {
	hooks := make([]*CommitHook, len(rpbHooks))
	for i, hook := range rpbHooks {
		commitHook := &CommitHook{
			Name: string(hook.Name),
		}
		if hook.Modfun != nil {
			commitHook.ModFun = &ModFun{
				Module:   string(hook.Modfun.Module),
				Function: string(hook.Modfun.Function),
			}
		}
		hooks[i] = commitHook
	}
	return hooks
}

// ModFun is used when fetching or updating LinkFun or ChashKeyfun bucket properties on Riak
type ModFun struct {
	Module   string
	Function string
}

func getFunFrom(rpbModFun *rpbRiak.RpbModFun) *ModFun {
	var modFun *ModFun
	if rpbModFun == nil {
		modFun = nil
	} else {
		modFun = &ModFun{
			Module:   string(rpbModFun.Module),
			Function: string(rpbModFun.Function),
		}
	}
	return modFun
}

// FetchBucketPropsResponse contains the response data for both
// FetchBucketPropsCommand and FetchBucketTypePropsCommand
type FetchBucketPropsResponse struct {
	NVal          uint32
	AllowMult     bool
	LastWriteWins bool
	HasPrecommit  bool
	HasPostcommit bool
	OldVClock     uint32
	YoungVClock   uint32
	BigVClock     uint32
	SmallVClock   uint32
	R             uint32
	Pr            uint32
	W             uint32
	Pw            uint32
	Dw            uint32
	Rw            uint32
	BasicQuorum   bool
	NotFoundOk    bool
	Search        bool
	Consistent    bool
	Repl          ReplMode
	Backend       string
	SearchIndex   string
	DataType      string
	PreCommit     []*CommitHook
	PostCommit    []*CommitHook
	ChashKeyFun   *ModFun
	LinkFun       *ModFun
	HllPrecision  uint32
}

func processRpbGetBucketResp(rsp *rpbRiak.RpbGetBucketResp) *FetchBucketPropsResponse {
	rpbBucketProps := rsp.GetProps()
	response := &FetchBucketPropsResponse{
		NVal:          rpbBucketProps.GetNVal(),
		AllowMult:     rpbBucketProps.GetAllowMult(),
		LastWriteWins: rpbBucketProps.GetLastWriteWins(),
		HasPrecommit:  rpbBucketProps.GetHasPrecommit(),
		HasPostcommit: rpbBucketProps.GetHasPostcommit(),
		OldVClock:     rpbBucketProps.GetOldVclock(),
		YoungVClock:   rpbBucketProps.GetYoungVclock(),
		BigVClock:     rpbBucketProps.GetBigVclock(),
		SmallVClock:   rpbBucketProps.GetSmallVclock(),
		R:             rpbBucketProps.GetR(),
		Pr:            rpbBucketProps.GetPr(),
		W:             rpbBucketProps.GetW(),
		Pw:            rpbBucketProps.GetPw(),
		Dw:            rpbBucketProps.GetDw(),
		Rw:            rpbBucketProps.GetRw(),
		BasicQuorum:   rpbBucketProps.GetBasicQuorum(),
		NotFoundOk:    rpbBucketProps.GetNotfoundOk(),
		Search:        rpbBucketProps.GetSearch(),
		Consistent:    rpbBucketProps.GetConsistent(),
		Repl:          ReplMode(rpbBucketProps.GetRepl()),
		Backend:       string(rpbBucketProps.GetBackend()),
		SearchIndex:   string(rpbBucketProps.GetSearchIndex()),
		DataType:      string(rpbBucketProps.GetDatatype()),
	}

	if rpbBucketProps.GetHasPrecommit() {
		response.PreCommit = getHooksFrom(rpbBucketProps.Precommit)
	}
	if rpbBucketProps.GetHasPostcommit() {
		response.PostCommit = getHooksFrom(rpbBucketProps.Postcommit)
	}
	if rpbBucketProps.ChashKeyfun != nil {
		response.ChashKeyFun = getFunFrom(rpbBucketProps.ChashKeyfun)
	}
	if rpbBucketProps.Linkfun != nil {
		response.LinkFun = getFunFrom(rpbBucketProps.Linkfun)
	}
	if rpbBucketProps.HllPrecision != nil {
		response.HllPrecision = rpbBucketProps.GetHllPrecision()
	}
	return response
}

// FetchBucketTypePropsCommand is used to fetch the active / non-default properties for a bucket type
type FetchBucketTypePropsCommand struct {
	commandImpl
	retryableCommandImpl
	Response *FetchBucketPropsResponse
	protobuf *rpbRiak.RpbGetBucketTypeReq
}

func (cmd *FetchBucketTypePropsCommand) Name() string {
	return cmd.getName("FetchBucketTypeProps")
}

func (cmd *FetchBucketTypePropsCommand) constructPbRequest() (proto.Message, error) {
	return cmd.protobuf, nil
}

func (cmd *FetchBucketTypePropsCommand) onSuccess(msg proto.Message) error {
	cmd.success = true
	if msg == nil {
		cmd.success = false
	} else {
		if rpbGetBucketResp, ok := msg.(*rpbRiak.RpbGetBucketResp); ok {
			cmd.Response = processRpbGetBucketResp(rpbGetBucketResp)
		} else {
			return fmt.Errorf("[FetchBucketTypePropsCommand] could not convert %v to RpbGetBucketResp", reflect.TypeOf(msg))
		}
	}
	return nil
}

func (cmd *FetchBucketTypePropsCommand) getRequestCode() byte {
	return rpbCode_RpbGetBucketTypeReq
}

func (cmd *FetchBucketTypePropsCommand) getResponseCode() byte {
	return rpbCode_RpbGetBucketResp
}

func (cmd *FetchBucketTypePropsCommand) getResponseProtobufMessage() proto.Message {
	return &rpbRiak.RpbGetBucketResp{}
}

// FetchBucketTypePropsCommandBuilder type is required for creating new instances of FetchBucketTypePropsCommand
//
//	command, err := NewFetchBucketTypePropsCommandBuilder().
//		WithBucketType("myBucketType").
//		Build()
type FetchBucketTypePropsCommandBuilder struct {
	protobuf *rpbRiak.RpbGetBucketTypeReq
}

// NewFetchBucketTypePropsCommandBuilder is a factory function for generating the command builder struct
func NewFetchBucketTypePropsCommandBuilder() *FetchBucketTypePropsCommandBuilder {
	builder := &FetchBucketTypePropsCommandBuilder{protobuf: &rpbRiak.RpbGetBucketTypeReq{}}
	return builder
}

// WithBucketType sets the bucket-type to be used by the command. If omitted, 'default' is used
func (builder *FetchBucketTypePropsCommandBuilder) WithBucketType(bucketType string) *FetchBucketTypePropsCommandBuilder {
	builder.protobuf.Type = []byte(bucketType)
	return builder
}

// Build validates the configuration options provided then builds the command
func (builder *FetchBucketTypePropsCommandBuilder) Build() (Command, error) {
	if builder.protobuf == nil {
		panic("builder.protobuf must not be nil")
	}
	if err := validateLocatable(builder.protobuf); err != nil {
		return nil, err
	}
	return &FetchBucketTypePropsCommand{protobuf: builder.protobuf}, nil
}

// FetchBucketPropsCommand is used to fetch the active / non-default properties for a bucket
type FetchBucketPropsCommand struct {
	commandImpl
	retryableCommandImpl
	Response *FetchBucketPropsResponse
	protobuf *rpbRiak.RpbGetBucketReq
}

func (cmd *FetchBucketPropsCommand) Name() string {
	return cmd.getName("FetchBucketProps")
}

func (cmd *FetchBucketPropsCommand) constructPbRequest() (proto.Message, error) {
	return cmd.protobuf, nil
}

func (cmd *FetchBucketPropsCommand) onSuccess(msg proto.Message) error {
	cmd.success = true
	if msg == nil {
		cmd.success = false
	} else {
		if rpbGetBucketResp, ok := msg.(*rpbRiak.RpbGetBucketResp); ok {
			cmd.Response = processRpbGetBucketResp(rpbGetBucketResp)
		} else {
			return fmt.Errorf("[FetchBucketPropsCommand] could not convert %v to RpbGetBucketResp", reflect.TypeOf(msg))
		}
	}
	return nil
}

func (cmd *FetchBucketPropsCommand) getRequestCode() byte {
	return rpbCode_RpbGetBucketReq
}

func (cmd *FetchBucketPropsCommand) getResponseCode() byte {
	return rpbCode_RpbGetBucketResp
}

func (cmd *FetchBucketPropsCommand) getResponseProtobufMessage() proto.Message {
	return &rpbRiak.RpbGetBucketResp{}
}

// FetchBucketPropsCommandBuilder type is required for creating new instances of FetchBucketPropsCommand
//
//	command, err := NewFetchBucketPropsCommandBuilder().
//		WithBucketType("myBucketType").
//		WithBucket("myBucket").
//		Build()
type FetchBucketPropsCommandBuilder struct {
	protobuf *rpbRiak.RpbGetBucketReq
}

// NewFetchBucketPropsCommandBuilder is a factory function for generating the command builder struct
func NewFetchBucketPropsCommandBuilder() *FetchBucketPropsCommandBuilder {
	builder := &FetchBucketPropsCommandBuilder{protobuf: &rpbRiak.RpbGetBucketReq{}}
	return builder
}

// WithBucketType sets the bucket-type to be used by the command. If omitted, 'default' is used
func (builder *FetchBucketPropsCommandBuilder) WithBucketType(bucketType string) *FetchBucketPropsCommandBuilder {
	builder.protobuf.Type = []byte(bucketType)
	return builder
}

// WithBucket sets the bucket to be used by the command
func (builder *FetchBucketPropsCommandBuilder) WithBucket(bucket string) *FetchBucketPropsCommandBuilder {
	builder.protobuf.Bucket = []byte(bucket)
	return builder
}

// Build validates the configuration options provided then builds the command
func (builder *FetchBucketPropsCommandBuilder) Build() (Command, error) {
	if builder.protobuf == nil {
		panic("builder.protobuf must not be nil")
	}
	if err := validateLocatable(builder.protobuf); err != nil {
		return nil, err
	}
	return &FetchBucketPropsCommand{protobuf: builder.protobuf}, nil
}

// StoreBucketTypePropsCommand is used to store changes to a bucket type's properties
type StoreBucketTypePropsCommand struct {
	commandImpl
	retryableCommandImpl
	protobuf *rpbRiak.RpbSetBucketTypeReq
}

// Name identifies this command
func (cmd *StoreBucketTypePropsCommand) Name() string {
	return cmd.getName("StoreBucketTypeProps")
}

func (cmd *StoreBucketTypePropsCommand) constructPbRequest() (proto.Message, error) {
	return cmd.protobuf, nil
}

func (cmd *StoreBucketTypePropsCommand) onSuccess(msg proto.Message) error {
	cmd.success = true
	return nil
}

func (cmd *StoreBucketTypePropsCommand) getRequestCode() byte {
	return rpbCode_RpbSetBucketTypeReq
}

func (cmd *StoreBucketTypePropsCommand) getResponseCode() byte {
	return rpbCode_RpbSetBucketResp
}

func (cmd *StoreBucketTypePropsCommand) getResponseProtobufMessage() proto.Message {
	return nil
}

// StoreBucketPropsCommand is used to store changes to a buckets properties
type StoreBucketPropsCommand struct {
	commandImpl
	retryableCommandImpl
	protobuf *rpbRiak.RpbSetBucketReq
}

// Name identifies this command
func (cmd *StoreBucketPropsCommand) Name() string {
	return cmd.getName("StoreBucketProps")
}

func (cmd *StoreBucketPropsCommand) constructPbRequest() (proto.Message, error) {
	return cmd.protobuf, nil
}

func (cmd *StoreBucketPropsCommand) onSuccess(msg proto.Message) error {
	cmd.success = true
	return nil
}

func (cmd *StoreBucketPropsCommand) getRequestCode() byte {
	return rpbCode_RpbSetBucketReq
}

func (cmd *StoreBucketPropsCommand) getResponseCode() byte {
	return rpbCode_RpbSetBucketResp
}

func (cmd *StoreBucketPropsCommand) getResponseProtobufMessage() proto.Message {
	return nil
}

// StoreBucketTypePropsCommandBuilder type is required for creating new instances of StoreBucketTypePropsCommand
//
//	command, err := NewStoreBucketTypePropsCommandBuilder().
//		WithBucketType("myBucketType").
//		WithAllowMult(true).
//		Build()
type StoreBucketTypePropsCommandBuilder struct {
	protobuf *rpbRiak.RpbSetBucketTypeReq
	props    *rpbRiak.RpbBucketProps
}

// NewStoreBucketTypePropsCommandBuilder is a factory function for generating the command builder struct
func NewStoreBucketTypePropsCommandBuilder() *StoreBucketTypePropsCommandBuilder {
	props := &rpbRiak.RpbBucketProps{}
	protobuf := &rpbRiak.RpbSetBucketTypeReq{
		Props: props,
	}
	builder := &StoreBucketTypePropsCommandBuilder{protobuf: protobuf, props: props}
	return builder
}

// WithBucketType sets the bucket-type to be used by the command. If omitted, 'default' is used
func (builder *StoreBucketTypePropsCommandBuilder) WithBucketType(bucketType string) *StoreBucketTypePropsCommandBuilder {
	builder.protobuf.Type = []byte(bucketType)
	return builder
}

// WithNVal sets the number of times this command operation is replicated in the Cluster. If
// omitted, the ring default is used.
//
// See http://basho.com/posts/technical/riaks-config-behaviors-part-2/
func (builder *StoreBucketTypePropsCommandBuilder) WithNVal(nval uint32) *StoreBucketTypePropsCommandBuilder {
	builder.props.NVal = &nval
	return builder
}

// WithAllowMult sets whether or not to allow Riak to store siblings of an object when writes conflict
func (builder *StoreBucketTypePropsCommandBuilder) WithAllowMult(allowMult bool) *StoreBucketTypePropsCommandBuilder {
	builder.props.AllowMult = &allowMult
	return builder
}

// WithLastWriteWins sets whether Riak should resolve conflicts using timestamp (most recent wins)
func (builder *StoreBucketTypePropsCommandBuilder) WithLastWriteWins(lww bool) *StoreBucketTypePropsCommandBuilder {
	builder.props.LastWriteWins = &lww
	return builder
}

// WithOldVClock sets the old_vclock value representing an epoch time value
func (builder *StoreBucketTypePropsCommandBuilder) WithOldVClock(oldVClock uint32) *StoreBucketTypePropsCommandBuilder {
	builder.props.OldVclock = &oldVClock
	return builder
}

// WithYoungVClock sets the old_vclock value representing an epoch time value
func (builder *StoreBucketTypePropsCommandBuilder) WithYoungVClock(youngVClock uint32) *StoreBucketTypePropsCommandBuilder {
	builder.props.YoungVclock = &youngVClock
	return builder
}

// WithBigVClock sets the big_vclock value representing an epoch time value
func (builder *StoreBucketTypePropsCommandBuilder) WithBigVClock(bigVClock uint32) *StoreBucketTypePropsCommandBuilder {
	builder.props.BigVclock = &bigVClock
	return builder
}

// WithSmallVClock sets the old_vclock value representing an epoch time value
func (builder *StoreBucketTypePropsCommandBuilder) WithSmallVClock(smallVClock uint32) *StoreBucketTypePropsCommandBuilder {
	builder.props.SmallVclock = &smallVClock
	return builder
}

// WithR sets the number of nodes that must report back a successful read in order for the
// command operation to be considered a success by Riak.
//
// See http://basho.com/posts/technical/riaks-config-behaviors-part-2/
func (builder *StoreBucketTypePropsCommandBuilder) WithR(r uint32) *StoreBucketTypePropsCommandBuilder {
	builder.props.R = &r
	return builder
}

// WithPr sets the number of primary nodes (N) that must be read from in order for the command
// operation to be considered a success by Riak.
//
// See http://basho.com/posts/technical/riaks-config-behaviors-part-2/
func (builder *StoreBucketTypePropsCommandBuilder) WithPr(pr uint32) *StoreBucketTypePropsCommandBuilder {
	builder.props.Pr = &pr
	return builder
}

// WithW sets the number of nodes that must report back a successful write in order for the
// command operation to be considered a success by Riak.
//
// See http://basho.com/posts/technical/riaks-config-behaviors-part-2/
func (builder *StoreBucketTypePropsCommandBuilder) WithW(w uint32) *StoreBucketTypePropsCommandBuilder {
	builder.props.W = &w
	return builder
}

// WithPw sets the number of primary nodes (N) that must report back a successful write in order for
// the command operation to be considered a success by Riak.
//
// See http://basho.com/posts/technical/riaks-config-behaviors-part-2/
func (builder *StoreBucketTypePropsCommandBuilder) WithPw(pw uint32) *StoreBucketTypePropsCommandBuilder {
	builder.props.Pw = &pw
	return builder
}

// WithDw (durable writes) sets the number of nodes that must report back a successful write to
// backend storage in order for the command operation to be considered a success by Riak. If
// omitted, the bucket default is used.
//
// See http://basho.com/posts/technical/riaks-config-behaviors-part-2/
func (builder *StoreBucketTypePropsCommandBuilder) WithDw(dw uint32) *StoreBucketTypePropsCommandBuilder {
	builder.props.Dw = &dw
	return builder
}

// WithRw (delete quorum) sets the number of nodes that must report back a successful delete to
// backend storage in order for the command operation to be considered a success by Riak. It
// represents the read and write operations that are completed internal to Riak to complete a delete.
//
// See http://basho.com/posts/technical/riaks-config-behaviors-part-2/
func (builder *StoreBucketTypePropsCommandBuilder) WithRw(rw uint32) *StoreBucketTypePropsCommandBuilder {
	builder.props.Rw = &rw
	return builder
}

// WithBasicQuorum sets basic_quorum, whether to return early in some failure cases (eg. when r=1
// and you get 2 errors and a success basic_quorum=true would return an error)
//
// See http://basho.com/posts/technical/riaks-config-behaviors-part-3/
func (builder *StoreBucketTypePropsCommandBuilder) WithBasicQuorum(basicQuorum bool) *StoreBucketTypePropsCommandBuilder {
	builder.props.BasicQuorum = &basicQuorum
	return builder
}

// WithNotFoundOk sets notfound_ok, whether to treat notfounds as successful reads for the purposes
// of R
//
// See http://basho.com/posts/technical/riaks-config-behaviors-part-3/
func (builder *StoreBucketTypePropsCommandBuilder) WithNotFoundOk(notFoundOk bool) *StoreBucketTypePropsCommandBuilder {
	builder.props.NotfoundOk = &notFoundOk
	return builder
}

// WithSearch enables / disables search features for this bucket
func (builder *StoreBucketTypePropsCommandBuilder) WithSearch(search bool) *StoreBucketTypePropsCommandBuilder {
	builder.props.Search = &search
	return builder
}

// WithBackend sets the backend to be used for this bucket
func (builder *StoreBucketTypePropsCommandBuilder) WithBackend(backend string) *StoreBucketTypePropsCommandBuilder {
	builder.props.Backend = []byte(backend)
	return builder
}

// WithSearchIndex sets a searchIndex to be used on the bucket
func (builder *StoreBucketTypePropsCommandBuilder) WithSearchIndex(searchIndex string) *StoreBucketTypePropsCommandBuilder {
	builder.props.SearchIndex = []byte(searchIndex)
	return builder
}

// AddPreCommit allows you to attach a precommit hook to the bucket
//
// See http://docs.basho.com/riak/latest/dev/using/commit-hooks/
func (builder *StoreBucketTypePropsCommandBuilder) AddPreCommit(commitHook *CommitHook) *StoreBucketTypePropsCommandBuilder {
	rpbCommitHook := toRpbCommitHook(commitHook)
	builder.props.Precommit = addCommitHookTo(builder.props.Precommit, rpbCommitHook)
	return builder
}

// AddPostCommit allows you to attach a postcommit hook to the bucket
//
// See http://docs.basho.com/riak/latest/dev/using/commit-hooks/
func (builder *StoreBucketTypePropsCommandBuilder) AddPostCommit(commitHook *CommitHook) *StoreBucketTypePropsCommandBuilder {
	rpbCommitHook := toRpbCommitHook(commitHook)
	builder.props.Postcommit = addCommitHookTo(builder.props.Postcommit, rpbCommitHook)
	return builder
}

// WithChashKeyFun sets the chash_keyfun property on the bucket which allows custom hashing functions
// Please note, this is an advanced feature, only use with caution
func (builder *StoreBucketTypePropsCommandBuilder) WithChashKeyFun(val *ModFun) *StoreBucketTypePropsCommandBuilder {
	builder.props.ChashKeyfun = &rpbRiak.RpbModFun{
		Module:   []byte(val.Module),
		Function: []byte(val.Function),
	}
	return builder
}

// WithHllPrecision sets the number of bits to use in the Hyperloglog data type precision.
// Valid values are [4 - 16] inclusive, default is 14 on new buckets.
// NOTE: When changing precision, it may only be reduced from
// it's current value, and never increased.
func (builder *StoreBucketTypePropsCommandBuilder) WithHllPrecision(precision uint32) *StoreBucketTypePropsCommandBuilder {
	builder.props.HllPrecision = &precision
	return builder
}

// Build validates the configuration options provided then builds the command
func (builder *StoreBucketTypePropsCommandBuilder) Build() (Command, error) {
	if builder.protobuf == nil {
		panic("builder.protobuf must not be nil")
	}
	if err := validateLocatable(builder.protobuf); err != nil {
		return nil, err
	}
	return &StoreBucketTypePropsCommand{protobuf: builder.protobuf}, nil
}

// StoreBucketPropsCommandBuilder type is required for creating new instances of StoreBucketPropsCommand
//
//	command := NewStoreBucketPropsCommandBuilder().
//		WithBucketType("myBucketType").
//		WithBucket("myBucket").
//		WithAllowMult(true).
//		Build()
type StoreBucketPropsCommandBuilder struct {
	protobuf *rpbRiak.RpbSetBucketReq
	props    *rpbRiak.RpbBucketProps
}

// NewStoreBucketPropsCommandBuilder is a factory function for generating the command builder struct
func NewStoreBucketPropsCommandBuilder() *StoreBucketPropsCommandBuilder {
	props := &rpbRiak.RpbBucketProps{}
	protobuf := &rpbRiak.RpbSetBucketReq{
		Props: props,
	}
	builder := &StoreBucketPropsCommandBuilder{protobuf: protobuf, props: props}
	return builder
}

// WithBucketType sets the bucket-type to be used by the command. If omitted, 'default' is used
func (builder *StoreBucketPropsCommandBuilder) WithBucketType(bucketType string) *StoreBucketPropsCommandBuilder {
	builder.protobuf.Type = []byte(bucketType)
	return builder
}

// WithBucket sets the bucket to be used by the command
func (builder *StoreBucketPropsCommandBuilder) WithBucket(bucket string) *StoreBucketPropsCommandBuilder {
	builder.protobuf.Bucket = []byte(bucket)
	return builder
}

// WithNVal sets the number of times this command operation is replicated in the Cluster. If
// omitted, the ring default is used.
//
// See http://basho.com/posts/technical/riaks-config-behaviors-part-2/
func (builder *StoreBucketPropsCommandBuilder) WithNVal(nval uint32) *StoreBucketPropsCommandBuilder {
	builder.props.NVal = &nval
	return builder
}

// WithAllowMult sets whether or not to allow Riak to store siblings of an object when writes conflict
func (builder *StoreBucketPropsCommandBuilder) WithAllowMult(allowMult bool) *StoreBucketPropsCommandBuilder {
	builder.props.AllowMult = &allowMult
	return builder
}

// WithLastWriteWins sets whether Riak should resolve conflicts using timestamp (most recent wins)
func (builder *StoreBucketPropsCommandBuilder) WithLastWriteWins(lww bool) *StoreBucketPropsCommandBuilder {
	builder.props.LastWriteWins = &lww
	return builder
}

// WithOldVClock sets the old_vclock value representing an epoch time value
func (builder *StoreBucketPropsCommandBuilder) WithOldVClock(oldVClock uint32) *StoreBucketPropsCommandBuilder {
	builder.props.OldVclock = &oldVClock
	return builder
}

// WithYoungVClock sets the old_vclock value representing an epoch time value
func (builder *StoreBucketPropsCommandBuilder) WithYoungVClock(youngVClock uint32) *StoreBucketPropsCommandBuilder {
	builder.props.YoungVclock = &youngVClock
	return builder
}

// WithBigVClock sets the big_vclock value representing an epoch time value
func (builder *StoreBucketPropsCommandBuilder) WithBigVClock(bigVClock uint32) *StoreBucketPropsCommandBuilder {
	builder.props.BigVclock = &bigVClock
	return builder
}

// WithSmallVClock sets the old_vclock value representing an epoch time value
func (builder *StoreBucketPropsCommandBuilder) WithSmallVClock(smallVClock uint32) *StoreBucketPropsCommandBuilder {
	builder.props.SmallVclock = &smallVClock
	return builder
}

// WithR sets the number of nodes that must report back a successful read in order for the
// command operation to be considered a success by Riak.
//
// See http://basho.com/posts/technical/riaks-config-behaviors-part-2/
func (builder *StoreBucketPropsCommandBuilder) WithR(r uint32) *StoreBucketPropsCommandBuilder {
	builder.props.R = &r
	return builder
}

// WithPr sets the number of primary nodes (N) that must be read from in order for the command
// operation to be considered a success by Riak.
//
// See http://basho.com/posts/technical/riaks-config-behaviors-part-2/
func (builder *StoreBucketPropsCommandBuilder) WithPr(pr uint32) *StoreBucketPropsCommandBuilder {
	builder.props.Pr = &pr
	return builder
}

// WithW sets the number of nodes that must report back a successful write in order for the
// command operation to be considered a success by Riak.
//
// See http://basho.com/posts/technical/riaks-config-behaviors-part-2/
func (builder *StoreBucketPropsCommandBuilder) WithW(w uint32) *StoreBucketPropsCommandBuilder {
	builder.props.W = &w
	return builder
}

// WithPw sets the number of primary nodes (N) that must report back a successful write in order for
// the command operation to be considered a success by Riak.
//
// See http://basho.com/posts/technical/riaks-config-behaviors-part-2/
func (builder *StoreBucketPropsCommandBuilder) WithPw(pw uint32) *StoreBucketPropsCommandBuilder {
	builder.props.Pw = &pw
	return builder
}

// WithDw (durable writes) sets the number of nodes that must report back a successful write to
// backend storage in order for the command operation to be considered a success by Riak. If
// omitted, the bucket default is used.
//
// See http://basho.com/posts/technical/riaks-config-behaviors-part-2/
func (builder *StoreBucketPropsCommandBuilder) WithDw(dw uint32) *StoreBucketPropsCommandBuilder {
	builder.props.Dw = &dw
	return builder
}

// WithRw (delete quorum) sets the number of nodes that must report back a successful delete to
// backend storage in order for the command operation to be considered a success by Riak. It
// represents the read and write operations that are completed internal to Riak to complete a delete.
//
// See http://basho.com/posts/technical/riaks-config-behaviors-part-2/
func (builder *StoreBucketPropsCommandBuilder) WithRw(rw uint32) *StoreBucketPropsCommandBuilder {
	builder.props.Rw = &rw
	return builder
}

// WithBasicQuorum sets basic_quorum, whether to return early in some failure cases (eg. when r=1
// and you get 2 errors and a success basic_quorum=true would return an error)
//
// See http://basho.com/posts/technical/riaks-config-behaviors-part-3/
func (builder *StoreBucketPropsCommandBuilder) WithBasicQuorum(basicQuorum bool) *StoreBucketPropsCommandBuilder {
	builder.props.BasicQuorum = &basicQuorum
	return builder
}

// WithNotFoundOk sets notfound_ok, whether to treat notfounds as successful reads for the purposes
// of R
//
// See http://basho.com/posts/technical/riaks-config-behaviors-part-3/
func (builder *StoreBucketPropsCommandBuilder) WithNotFoundOk(notFoundOk bool) *StoreBucketPropsCommandBuilder {
	builder.props.NotfoundOk = &notFoundOk
	return builder
}

// WithSearch enables / disables search features for this bucket
func (builder *StoreBucketPropsCommandBuilder) WithSearch(search bool) *StoreBucketPropsCommandBuilder {
	builder.props.Search = &search
	return builder
}

// WithBackend sets the backend to be used for this bucket
func (builder *StoreBucketPropsCommandBuilder) WithBackend(backend string) *StoreBucketPropsCommandBuilder {
	builder.props.Backend = []byte(backend)
	return builder
}

// WithSearchIndex sets a searchIndex to be used on the bucket
func (builder *StoreBucketPropsCommandBuilder) WithSearchIndex(searchIndex string) *StoreBucketPropsCommandBuilder {
	builder.props.SearchIndex = []byte(searchIndex)
	return builder
}

// AddPreCommit allows you to attach a precommit hook to the bucket
//
// See http://docs.basho.com/riak/latest/dev/using/commit-hooks/
func (builder *StoreBucketPropsCommandBuilder) AddPreCommit(commitHook *CommitHook) *StoreBucketPropsCommandBuilder {
	rpbCommitHook := toRpbCommitHook(commitHook)
	builder.props.Precommit = addCommitHookTo(builder.props.Precommit, rpbCommitHook)
	return builder
}

// AddPostCommit allows you to attach a postcommit hook to the bucket
//
// See http://docs.basho.com/riak/latest/dev/using/commit-hooks/
func (builder *StoreBucketPropsCommandBuilder) AddPostCommit(commitHook *CommitHook) *StoreBucketPropsCommandBuilder {
	rpbCommitHook := toRpbCommitHook(commitHook)
	builder.props.Postcommit = addCommitHookTo(builder.props.Postcommit, rpbCommitHook)
	return builder
}

// WithChashKeyFun sets the chash_keyfun property on the bucket which allows custom hashing functions
// Please note, this is an advanced feature, only use with caution
func (builder *StoreBucketPropsCommandBuilder) WithChashKeyFun(val *ModFun) *StoreBucketPropsCommandBuilder {
	builder.props.ChashKeyfun = &rpbRiak.RpbModFun{
		Module:   []byte(val.Module),
		Function: []byte(val.Function),
	}
	return builder
}

// WithHllPrecision sets the number of bits to use in the Hyperloglog data type precision.
// Valid values are [4 - 16] inclusive, default is 14 on new buckets.
// NOTE: When changing precision, it may only be reduced from
// it's current value, and never increased.
func (builder *StoreBucketPropsCommandBuilder) WithHllPrecision(precision uint32) *StoreBucketPropsCommandBuilder {
	builder.props.HllPrecision = &precision
	return builder
}

// Build validates the configuration options provided then builds the command
func (builder *StoreBucketPropsCommandBuilder) Build() (Command, error) {
	if builder.protobuf == nil {
		panic("builder.protobuf must not be nil")
	}
	if err := validateLocatable(builder.protobuf); err != nil {
		return nil, err
	}
	return &StoreBucketPropsCommand{protobuf: builder.protobuf}, nil
}

// ResetBucketCommandBuilder is the command builder for ResetBucketCommand
type ResetBucketCommandBuilder struct {
	protobuf *rpbRiak.RpbResetBucketReq
}

// NewResetBucketCommandBuilder is a factory function for generating the command builder struct
func NewResetBucketCommandBuilder() *ResetBucketCommandBuilder {
	builder := &ResetBucketCommandBuilder{protobuf: &rpbRiak.RpbResetBucketReq{}}
	return builder
}

// WithBucketType sets the bucket-type to be used by the command. If omitted, 'default' is used
func (builder *ResetBucketCommandBuilder) WithBucketType(bucketType string) *ResetBucketCommandBuilder {
	builder.protobuf.Type = []byte(bucketType)
	return builder
}

// WithBucket sets the bucket to be used by the command
func (builder *ResetBucketCommandBuilder) WithBucket(bucket string) *ResetBucketCommandBuilder {
	builder.protobuf.Bucket = []byte(bucket)
	return builder
}

// Build validates the configuration options provided then builds the command
func (builder *ResetBucketCommandBuilder) Build() (Command, error) {
	if builder.protobuf == nil {
		panic("builder.protobuf must not be nil")
	}
	if err := validateLocatable(builder.protobuf); err != nil {
		return nil, err
	}
	return &ResetBucketCommand{protobuf: builder.protobuf}, nil
}

// ResetBucketCommand is used to reset the properties of a given bucket or bucket type
type ResetBucketCommand struct {
	commandImpl
	retryableCommandImpl
	protobuf *rpbRiak.RpbResetBucketReq
}

// Name identifies this command
func (cmd *ResetBucketCommand) Name() string {
	return cmd.getName("ResetBucket")
}

func (cmd *ResetBucketCommand) getRequestCode() byte {
	return rpbCode_RpbResetBucketReq
}

func (cmd *ResetBucketCommand) constructPbRequest() (msg proto.Message, err error) {
	return cmd.protobuf, nil
}

func (cmd *ResetBucketCommand) onSuccess(msg proto.Message) error {
	cmd.success = true
	return nil
}

func (cmd *ResetBucketCommand) getResponseCode() byte {
	return rpbCode_RpbResetBucketResp
}

func (cmd *ResetBucketCommand) getResponseProtobufMessage() proto.Message {
	return nil
}

func addCommitHookTo(rpbHooks []*rpbRiak.RpbCommitHook, rpbCommitHook *rpbRiak.RpbCommitHook) []*rpbRiak.RpbCommitHook {
	return append(rpbHooks, rpbCommitHook)
}

func toRpbCommitHook(commitHook *CommitHook) *rpbRiak.RpbCommitHook {
	rpbCommitHook := &rpbRiak.RpbCommitHook{
		Name: []byte(commitHook.Name),
	}
	if commitHook.ModFun != nil {
		rpbCommitHook.Modfun = &rpbRiak.RpbModFun{
			Module:   []byte(commitHook.ModFun.Module),
			Function: []byte(commitHook.ModFun.Function),
		}
	}
	return rpbCommitHook
}

type authCommand struct {
	commandImpl
	user     string
	password string
}

func (cmd *authCommand) Name() string {
	return cmd.getName("Auth")
}

func (cmd *authCommand) constructPbRequest() (msg proto.Message, err error) {
	return &rpbRiak.RpbAuthReq{
		User:     []byte(cmd.user),
		Password: []byte(cmd.password),
	}, nil
}

func (cmd *authCommand) onSuccess(msg proto.Message) error {
	cmd.success = true
	return nil
}

func (cmd *authCommand) getRequestCode() byte {
	return rpbCode_RpbAuthReq
}

func (cmd *authCommand) getResponseCode() byte {
	return rpbCode_RpbAuthResp
}

func (cmd *authCommand) getResponseProtobufMessage() proto.Message {
	return nil
}