structs.go

// Copyright 2012, Google Inc. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package actionnode

import (
	"fmt"

	"github.com/youtube/vitess/go/vt/key"
	myproto "github.com/youtube/vitess/go/vt/mysqlctl/proto"
	"github.com/youtube/vitess/go/vt/topo"
)

/*
This file defines all the payload structures for the ActionNode objects.

The naming conventions are:
- a structure used for Args only is suffixed by 'Args'.
- a structure used for Reply only is suffixed by 'Reply'.
- a structure used for both Args and Reply is suffixed by 'Data'.

Note it's OK to rename the structures as the type name is not saved in json.
*/

// tablet action node structures

type RestartSlaveData struct {
	ReplicationState *myproto.ReplicationState
	WaitPosition     *myproto.ReplicationPosition
	TimePromoted     int64 // used to verify replication - a row will be inserted with this timestamp
	Parent           topo.TabletAlias
	Force            bool
}

func (rsd *RestartSlaveData) String() string {
	return fmt.Sprintf("RestartSlaveData{ReplicationState:%#v WaitPosition:%#v TimePromoted:%v Parent:%v Force:%v}", rsd.ReplicationState, rsd.WaitPosition, rsd.TimePromoted, rsd.Parent, rsd.Force)
}

type SlaveWasRestartedArgs struct {
	Parent topo.TabletAlias
}

type SnapshotArgs struct {
	Concurrency int
	ServerMode  bool
}

type SnapshotReply struct {
	ParentAlias  topo.TabletAlias
	ManifestPath string

	// these two are only used for ServerMode=true full snapshot
	SlaveStartRequired bool
	ReadOnly           bool
}

type MultiSnapshotReply struct {
	ParentAlias   topo.TabletAlias
	ManifestPaths []string
}

type SnapshotSourceEndArgs struct {
	SlaveStartRequired bool
	ReadOnly           bool
}

type MultiSnapshotArgs struct {
	KeyRanges        []key.KeyRange
	Tables           []string
	ExcludeTables    []string
	Concurrency      int
	SkipSlaveRestart bool
	MaximumFilesize  uint64
}

type MultiRestoreArgs struct {
	SrcTabletAliases       []topo.TabletAlias
	Concurrency            int
	FetchConcurrency       int
	InsertTableConcurrency int
	FetchRetryCount        int
	Strategy               string
}

type ReserveForRestoreArgs struct {
	SrcTabletAlias topo.TabletAlias
}

type RestoreArgs struct {
	SrcTabletAlias        topo.TabletAlias
	SrcFilePath           string
	ParentAlias           topo.TabletAlias
	FetchConcurrency      int
	FetchRetryCount       int
	WasReserved           bool
	DontWaitForSlaveStart bool
}

// shard action node structures

type ApplySchemaShardArgs struct {
	MasterTabletAlias topo.TabletAlias
	Change            string
	Simple            bool
}

type SetShardServedTypesArgs struct {
	ServedTypes []topo.TabletType
}

type MigrateServedTypesArgs struct {
	ServedType topo.TabletType
}

// keyspace action node structures

type ApplySchemaKeyspaceArgs struct {
	Change string
	Simple bool
}

type MigrateServedFromArgs struct {
	ServedType topo.TabletType
}

// methods to build the shard action nodes

func ReparentShard(tabletAlias topo.TabletAlias) *ActionNode {
	return (&ActionNode{
		Action: SHARD_ACTION_REPARENT,
		Args:   &tabletAlias,
	}).SetGuid()
}

func ShardExternallyReparented(tabletAlias topo.TabletAlias) *ActionNode {
	return (&ActionNode{
		Action: SHARD_ACTION_EXTERNALLY_REPARENTED,
		Args:   &tabletAlias,
	}).SetGuid()
}

func RebuildShard() *ActionNode {
	return (&ActionNode{
		Action: SHARD_ACTION_REBUILD,
	}).SetGuid()
}

func CheckShard() *ActionNode {
	return (&ActionNode{
		Action: SHARD_ACTION_CHECK,
	}).SetGuid()
}

func ApplySchemaShard(masterTabletAlias topo.TabletAlias, change string, simple bool) *ActionNode {
	return (&ActionNode{
		Action: SHARD_ACTION_APPLY_SCHEMA,
		Args: &ApplySchemaShardArgs{
			MasterTabletAlias: masterTabletAlias,
			Change:            change,
			Simple:            simple,
		},
	}).SetGuid()
}

func SetShardServedTypes(servedTypes []topo.TabletType) *ActionNode {
	return (&ActionNode{
		Action: SHARD_ACTION_SET_SERVED_TYPES,
		Args: &SetShardServedTypesArgs{
			ServedTypes: servedTypes,
		},
	}).SetGuid()
}

func ShardMultiRestore(args *MultiRestoreArgs) *ActionNode {
	return (&ActionNode{
		Action: SHARD_ACTION_MULTI_RESTORE,
		Args:   args,
	}).SetGuid()
}

func MigrateServedTypes(servedType topo.TabletType) *ActionNode {
	return (&ActionNode{
		Action: SHARD_ACTION_MIGRATE_SERVED_TYPES,
		Args: &MigrateServedTypesArgs{
			ServedType: servedType,
		},
	}).SetGuid()
}

func UpdateShard() *ActionNode {
	return (&ActionNode{
		Action: SHARD_ACTION_UPDATE_SHARD,
	}).SetGuid()
}

// methods to build the keyspace action nodes

func RebuildKeyspace() *ActionNode {
	return (&ActionNode{
		Action: KEYSPACE_ACTION_REBUILD,
	}).SetGuid()
}

func SetKeyspaceShardingInfo() *ActionNode {
	return (&ActionNode{
		Action: KEYSPACE_ACTION_SET_SHARDING_INFO,
	}).SetGuid()
}

func ApplySchemaKeyspace(change string, simple bool) *ActionNode {
	return (&ActionNode{
		Action: KEYSPACE_ACTION_APPLY_SCHEMA,
		Args: &ApplySchemaKeyspaceArgs{
			Change: change,
			Simple: simple,
		},
	}).SetGuid()
}

func MigrateServedFrom(servedType topo.TabletType) *ActionNode {
	return (&ActionNode{
		Action: KEYSPACE_ACTION_MIGRATE_SERVED_FROM,
		Args: &MigrateServedFromArgs{
			ServedType: servedType,
		},
	}).SetGuid()
}

//methods to build the serving shard action nodes

func RebuildSrvShard() *ActionNode {
	return (&ActionNode{
		Action: SRV_SHARD_ACTION_REBUILD,
	}).SetGuid()
}